Why do you move file one by one in proc_images.py？

[castleKing1997]

If it is faster with

mv label+"*" label

?

[castleKing1997]

<-----

for datatype in ['train', 'val', 'test']:
    os.system('mkdir ' + datatype)

    with open(datatype + '.csv', 'r') as f:
        reader = csv.reader(f, delimiter=',')
        last_label = ''
        for i, row in enumerate(reader):
            if i == 0:  # skip the headers
                continue
            label = row[1]
            image_name = row[0]
            if label != last_label:
                cur_dir = datatype + '/' + label + '/'
                os.system('mkdir ' + cur_dir)
                last_label = label
            os.system('mv images/' + image_name + ' ' + cur_dir)

----->

for datatype in ['train', 'val', 'test']:
    os.system('mkdir ' + datatype)
    data=pd.read_csv(datatype + '.csv')
    label_set = set(data['label'])
    for label in label_set:
        cur_dir = datatype + '/' + label + '/'
        os.system('mkdir ' + cur_dir)
        os.system('mv images/' + label+"*" + ' ' + cur_dir)