kaggle2fastai dataset conventer

conventer.py

import pandas as pd
import os
from shutil import copyfile,move

labels = pd.read_csv('data/dogbreed/labels.csv')
classes = list(labels['breed'].unique())

source_path = 'data/dogbreed/old_train/'
path = 'data/dogbreed/train/'
valid_path = 'data/dogbreed/valid/'
for index, row in labels.iterrows():
    dst = path + row['breed'] + "/"
    fname =  row['id'] + '.jpg'
    if not os.path.exists(dst):
        os.makedirs(dst)
    
    copyfile(source_path + fname, dst + fname)
    
for cls in classes:
    parent_list = os.listdir(path + cls)
    count =0
    dst =  valid_path + cls + "/"
    if not os.path.exists(dst):
        os.makedirs(dst)
    for child in parent_list:
        if count < 20:
            move(path + cls + "/" + child, dst + child) 
        else:
            break
        count = count+1