Thanks for the repo. I'm preparing the dataset according to your prepare_data.py file. But I don't know where the root_dir, 'original_partition', mod + '.lst' files come from? Please help me!
this code from your file :
for mod in mode:
part_file = os.path.join(root_dir, 'original_partition', mod + '.lst')
with open(part_file)as fp:
for line in fp:
name = line.split('-')
if int(name[-1][:-1]) == 0:
anno_file = os.path.join(labels_dir, '-'.join(name[:2]) + '.xml')
full_ann_files.append(anno_file)
im_dir = os.path.join(images_dir, name[0], '-'.join(name[:2]))
im_dirs.append(im_dir)