Perf
Perf copied to clipboard
Where I can get the train_list.txt in Resnet50 experiment
I can't find train_list.txt in ImageNet dataset.
I generate by
def generate_train():
paths = os.walk(r'/root/data/train/')
i = 0
for path, dir_lst, file_lst in paths:
for dir in dir_lst:
dirpath = os.walk(r'/root/data/train/'+dir)
for dpath, dlist, f_lst in dirpath:
for file_name in f_lst:
print("train/{:s}/{:s} {:d}".format(dir,file_name,i))
i+=1