generate_train_txt_and_test_txt.py

import os, csv, time
import glob


def write_list_file(listfile, jpg_file):
    f = open(listfile, 'w')

    wd = os.getcwd()
    for filename in glob.glob("{}/JPEGImages/{}".format(wd, jpg_file)):
        print(filename)
        f.write(filename+"\n")

    f.close()

write_list_file('train.txt', '[0-5]*.jpg')
write_list_file('test.txt', '6*.jpg')