Create corresponding trainval.txt for existing dataset image files

create_txt.py

import os

path = os.getcwd()
dataAnnotated = os.listdir(path + '/JPEGImages')
dataNum = len(dataAnnotated)

ftrainval = open('trainval.txt', 'w')
info = ''
for name in dataAnnotated:
    info += str(name).strip(".jpg")
    info += '\n'
ftrainval.write(str(info))
ftrainval.close()
print('done!')