move images into sub directories for keras ImageDataGenerator

structure_keras_dir.py

import os
with open('../data/labels.csv','r') as labels:
    lines = [line.split(',') for line in labels.readlines()][1:]
    for line in lines:
        image = line[0]
        label = line[1].split('/')[-1].replace('\n','')
        if not os.path.exists('../data/train/'+label):
            os.mkdir('../data/train/'+label)
        os.rename('../data/train/'+image+'.jpg', '../data/train/'+label+'/'+image+'.jpg')