Perf icon indicating copy to clipboard operation
Perf copied to clipboard

Where I can get the train_list.txt in Resnet50 experiment

Open LukeLIN-web opened this issue 2 years ago • 1 comments

I can't find train_list.txt in ImageNet dataset.

LukeLIN-web avatar Jul 12 '22 07:07 LukeLIN-web

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

LukeLIN-web avatar Jul 13 '22 07:07 LukeLIN-web