def get_img(self, folders, phase):
record = []
for folder in folders:
current_root = os.path.join(self.image_root, phase, folder)
images = get_name(current_root, mode_folder=False)
for img in images:
record.append([os.path.join(current_root, img), self.deal_label(folder)])
return record
def __init__(self, args, ratio=0.8):
self.image_root = args.dataset_dir
self.all_image = get_name(self.image_root, mode_folder=False)
self.category = sorted(set([i[:i.find('_')] for i in self.all_image]))
for c_id, c in enumerate(self.category):
print(c_id, '\t', c)
self.ration = ratio
def __init__(self, args):
self.image_root = args.dataset_dir
self.category = get_name(self.image_root + "train/")
self.used_cat = self.category[:args.num_classes]