yolov3 训练自己的数据集之生成train.txt

from os.path import join
import os
img_dir = "images"
work_dir = os.getcwd()
print ("当前工作目录 : %s" % work_dir)
txt_file = open(work_dir+'/train.txt', 'w')
img_path = join(work_dir,img_dir)
print ("图片文件夹 :",img_path)
image_names = os.listdir('./'+ img_dir)
for file_name in image_names:
    if file_name.endswith(".jpg"):
        txt_file.write(img_path + "/"+ file_name+"\n")
txt_file.close()

work_dir是此py脚本的路径，比如E:\workspace\classify

img_dir 是数据集图片的路径，比如我把图片都放在了，E:\workspace\classify\images中

我这里直接把work_dir +“images”，就得到了img_dir，如果你的路径和不一样自行修改一下

然后遍历img_dir下的所有jpg图片，如果你是别的图片你把 if file_name.endswith(".jpg") 改一改，不要判断后缀了，把每张图片的绝对路径写在txt中。

注意一下，在win系统生成的txt,不能放在linux 中用，因为反斜杠的原因。