PaddleDetection训练自己的数据说明:
一、数据标注
使用labelme/labelimg标注VOC数据集
二、数据预整理
2.1 VOC数据集整理合并划分
import random
import os
# 生成train.txt和val.txt
random.seed(2020)
xml_dir = '/home/aistudio/work/cigarette_defect/Annotations'
img_dir = '/home/aistudio/work/cigarette_defect/JPEGImages'
path_list = list()
for img in os.listdir(img_dir):
img_path = os.path.join(img_dir,img)
xml_path = os.path.join(xml_dir,img.replace('jpg', 'xml'))
path_list.append((img_path, xml_path))
random.shuffle(path_li