YOLOv5训练自定义的烟火数据集和COCO2017数据集

一. 训练自定义数据集

1、准备数据集

(1)数据集具体结构内容如下：
yolov5/data/images存放训练的jpg图片
yolov5/data/annotations存放对应图片标注的xml文件
yolov5/data/ImageSets新建Main文件夹存放划分的训练集、测试集等文件

(2)运行voc2yolo5.py生成划分的训练集、测试集等文件

import os
import random 

xmlfilepath=r'annotations'
saveBasePath=r"ImageSets/Main/"

trainval_percent=0.66
train_percent=0.5

temp_xml = os.listdir(xmlfilepath)
total_xml = []
for xml in temp_xml:
  if xml.endswith(".xml"):
    	total_xml.append(xml)

num=len(total_xml)  
list=range(num)  
tv=int(num*trainval_percent)  
tr=int(tv*train_percent)  
trainval= random.sample(list,tv)  
train=random.sample(trainval,tr)  

print("train and val size",tv)
print("traub suze",tr)
ftrainval = open(os.path.join(saveBasePath,'tr