In [1]:

    

import sys
import glob
import os
import os.path
import random


    

In [2]:

    

TRAIN_TEXT_FILE = '/Users/Enkay/Documents/Viky/python/img-classification/train_shuffle.txt'

IMAGE_FOLDER = '/Users/Enkay/Documents/Viky/python/img-classification/train/'


    

In [4]:

    

fr = open(TRAIN_TEXT_FILE, 'w')

filenames_image = os.listdir(IMAGE_FOLDER)
random.shuffle(filenames_image)
for filename in filenames_image:
    if filename[0:3] == 'cat': # Class '0' for cat 
        fr.write(IMAGE_FOLDER + '/'+ filename + ' 0\n')
    elif filename[0:3] == 'dog': # c
        fr.write(IMAGE_FOLDER + '/'+ filename + ' 1\n')
fr.close()