PaddleDetection训练自己的数据集遇到的问题及解决办法

PaddleDetection训练自己的数据集遇到的问题及解决办法

1.使用labelImg标注工具标注自己的数据集得到.xml文件
2.自动创建train.txt、val.txt，这两个文件中储存训练集、验证集 图像的路径信息和对应的xml文件
参考https://blog.csdn.net/yzl819819/article/details/104336990的博客

import os
import random

train_precent=0.7
xml="VOC2012/Annotations"
save="VOC2012/ImageSets/Main"
total_xml=os.listdir(xml)

num=len(total_xml)
tr=int(num*train_precent)
train=range(0,tr)

ftrain=open("VOC2012/ImageSets/Main/train.txt","w")
ftest=open("VOC2012/ImageSets/Main/val.txt","w")

for i in range(num):
    name=total_xml[i][:-4]+"\n"
    if i in train:
        ftrain.write(name)
    else:
        ftest.write(name)

ftrain.close()
ftest