train_test_split.py

import os
from os.path import isfile
import random

annots = os.listdir("annotations")
random.shuffle(annots)
test_len = int(len(annots) * 0.1)
train_data = annots[test_len:]
test_data = annots[:test_len]
f = open('ImageSets/train_messi.txt', 'w')
for annot in train_data:
	name = annot[: annot.index('.xml')]
	png_image = 'images/' + name + '.png'
	if isfile(png_image):
		f.write(png_image + "\t" + "annotations/" + annot + "\n")

f.close()

f = open('ImageSets/test_messi.txt', 'w')
for annot in test_data:
	name = annot[: annot.index('.xml')]
	png_image = 'images/' + name + '.png'
	if isfile(png_image):
		f.write(png_image + "\t" + "annotations/" + annot + "\n")

f.close()