Creates train.txt and test.txt files

datasplit.py

import os
import random

imgspath = 'images' #path to images
path = '/content/gdrive/My Drive/darknet/build/darknet/x64/data/obj/'
images = []
for i in os.listdir(imgspath):
    temp = path+i
    images.append(temp)
# train and test split... adjust it if necessary
trainlen = round(len(images)*.80)
testlen = round(len(images)*.20)
#print('total, train, test dataset size -',trainlen+testlen,trainlen,testlen)
random.shuffle(images)
test = images[:testlen]
train = images[testlen:]

with open('train.txt', 'w') as f:
    for item in train:
        f.write("%s\n" % item)
with open('test.txt', 'w') as f:
    for item in test:
        f.write("%s\n" % item)