class TestDataset:
def __init__(self, opt, split='test', use_difficult=True):
self.opt = opt
self.db = VOCDataset(opt.voc_data_dir,split=split)
def __getitem__(self, idx):
ori_img, bbox, label, difficult = self.db.get_example(idx)
img = preprocess(ori_img)
return img, ori_img.shape[1:], bbox, label, difficult
def __len__(self):
return len(self.db)
class Dataset:
def __init__(self, opt):
self.opt = opt
#生成类的实例
self.db = VOCDataset(opt.voc_data_dir)
# 生成类的实例
self.tsf = Transform(opt.min_size, opt.max_size)
def __getitem__(self, idx):
#调用函数来获取一张图片的标注信息
ori_img, bbox, label, difficult = self.db.get_example(idx)
#
img, bbox, label, scale = self.tsf((ori_img, bbox, label))
return img.copy(), bbox.copy(), label.copy(), scale
def __len__(self):
return len(self.db)