Divide folder of images into train and test

train_test.py

def split_train_test(categories, PATH, percent_test=0.2):
    for category in categories:
        print(len(os.listdir(PATH+category)))
        files = os.listdir(PATH+category)
        for file in files:
            if np.random.rand(1) < percent_test:
                shutil.copyfile(PATH + category +"/" + file, PATH + "test/" +category +"/" + file)
            else:
                shutil.copyfile(PATH + category +"/" + file, PATH + "train/"+ category +"/" + file)