代码:https://github.com/dog-qiuqiu/FastestDet
一、构造数据集
数据集格式YOLO相同,每张图片对应一个txt标签文件。标签格式:“category cx cy wh”,category为类别id,cx, cy为归一化标签框中心点的坐标,w, h为归一化标签框的宽度和高度,
.txt标签文件内容示例如下:
0 0.51 0.32 0.0284 0.025
1 0.47 0.32 0.0625 0.1125
数据集文件结构如下:
images路径下分train和val两个文件夹,每个文件夹下:
如下为VOC转YOLO的代码:
import xml.etree.ElementTree as ET
import pickle
import os
from os