FastestDet---模型训练

代码: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
数据集文件结构如下:
FastestDet---模型训练_第1张图片
images路径下分train和val两个文件夹,每个文件夹下:
FastestDet---模型训练_第2张图片
如下为VOC转YOLO的代码:

import xml.etree.ElementTree as ET
import pickle
import os
from os 

你可能感兴趣的:(模型部署,目标检测,计算机视觉)