
YOLOV4训练自己数据集过程
发布日期:2021-05-14 10:12:02
浏览次数:12
分类:精选文章
本文共 780 字,大约阅读时间需要 2 分钟。
VOC格式训练指南
1. 基本配置
本文采用VOC格式进行训练,所有操作需在VOCdevkit文件夹下的VOC2007文件夹中完成。
2. 准备训练数据
在训练前,请将标签文件放置在 VOC2007/Annotation
文件夹中。使用 test.py
脚本生成对应的.txt文件。
3. 配置训练参数
打开 voc_annotation.py
脚本,修改 classes
参数为你定义的类别。请注意:
- 不使用中文标签
- 文件夹内不得有空格
示例类别配置:
classes = ["aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"]
4. 生成训练数据
运行上述脚本,会自动生成 2007_train.txt
文件。每行记录对应图片的路径及其真实框位置。
5. 创建分类文件
在 model_data
文件夹下新建一个.txt文件,文件内容为所需分类用逗号分隔。例如:
cat,dog
将该路径设置为 train.py
中的 classes_path
参数,示例设置:
classes_path = 'model_data/my_class.txt'
6. 开始训练
运行 train.py
脚本即可完成训练准备工作。训练完成后,可生成所需的训练数据集。
注意事项
- 运行前确保所有路径正确且不包含空格
- 不要使用中文标签
- 保持文件夹结构完整,避免多级目录
- 可根据实际需求修改训练参数
如有问题,请根据文档详细信息进行查阅和调整。
发表评论
最新留言
很好
[***.229.124.182]2025年04月18日 05时43分07秒
关于作者

喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
《web安全入门》(四)前端开发基础Javascript
2019-03-07
python中列表 元组 字典 集合的区别
2019-03-07
python struct 官方文档
2019-03-07
Android DEX加固方案与原理
2019-03-07
Android Retrofit2.0 上传单张图片和多张图片
2019-03-07
iOS_Runtime3_动态添加方法
2019-03-07
Leetcode第557题---翻转字符串中的单词
2019-03-07
Problem G. The Stones Game【取石子博弈 & 思维】
2019-03-07
Unable to execute dex: Multiple dex files
2019-03-07
Java多线程
2019-03-07
Unity监听日记
2019-03-07
AndroidStudio跳到错误位置
2019-03-07
木马开发的基本理论基础(五)
2019-03-07
openssl服务器证书操作
2019-03-07
expect 模拟交互 ftp 上传文件到指定目录下
2019-03-07
linux系统下双屏显示
2019-03-07
PDF.js —— vue项目中使用pdf.js显示pdf文件(流)
2019-03-07
我用wxPython搭建GUI量化系统之最小架构的运行
2019-03-07