当前位置:网站首页>yolov3训练自己的数据集(MMDetection)
yolov3训练自己的数据集(MMDetection)
2022-07-02 06:26:00 【chenf0】
用FasterRcnn训练了自己标注的数据集Voc格式,现在想用yolo来训练一下,修改了yolo文件内容,打算直接用yolo训练voc格式的数据,出现了一点问题 ,因为比较着急,就没有再详细研究。
MMDetection中大多训练模型为coco格式设计,计划把 voc格式转换为coco格式,也方便以后其他模型的训练。
1.voc格式转换为coco格式
可参考:
https://github.com/Stephenfang51/VOC_to_COCO
2.coco相关文件的修改
可参考
【mmdetection】使用自定义的coco格式数据集进行训练及测试
(1)定义数据种类(mmdetection/mmdet/datasets/coco.py),把CLASSES的那个tuple改为自己数据集对应的种类tuple即可。
CLASSES = ('Other Car', 'Taxi')
(2)修改coco_classes数据集类别(mmdetection/mmdet/core/evaluation/class_names.py)
def coco_classes():
return [
'Other Car', 'Taxi'
]
(3)修改使用模型model字典中的num_classes
num_classes=2,#类别数
3.训练
python tools/train.py configs/yolo/yolov3_d53_mstrain-608_273e_coco.py
边栏推荐
- Message queue fnd in Oracle EBS_ msg_ pub、fnd_ Application of message in pl/sql
- Data warehouse model fact table model design
- [model distillation] tinybert: distilling Bert for natural language understanding
- ssm超市订单管理系统
- 腾讯机试题
- Using MATLAB to realize: Jacobi, Gauss Seidel iteration
- ERNIE1.0 与 ERNIE2.0 论文解读
- 常见CNN网络创新点
- Two dimensional array de duplication in PHP
- 矩阵的Jordan分解实例
猜你喜欢

【Ranking】Pre-trained Language Model based Ranking in Baidu Search

A slide with two tables will help you quickly understand the target detection

ssm+mysql实现进销存系统

【信息检索导论】第二章 词项词典与倒排记录表

点云数据理解(PointNet实现第3步)

Regular expressions in MySQL

Implementation of purchase, sales and inventory system with ssm+mysql

Ding Dong, here comes the redis om object mapping framework

常见CNN网络创新点

传统目标检测笔记1__ Viola Jones
随机推荐
华为机试题
【信息检索导论】第三章 容错式检索
【Torch】最简洁logging使用指南
Delete the contents under the specified folder in PHP
机器学习理论学习:感知机
ssm垃圾分类管理系统
view的绘制机制(二)
离线数仓和bi开发的实践和思考
The first quickapp demo
CRP implementation methodology
实现接口 Interface Iterable<T>
读《敏捷整洁之道:回归本源》后感
Oracle 11g uses ords+pljson to implement JSON_ Table effect
[introduction to information retrieval] Chapter 3 fault tolerant retrieval
Oracle EBs and apex integrated login and principle analysis
Ding Dong, here comes the redis om object mapping framework
【Ranking】Pre-trained Language Model based Ranking in Baidu Search
Oracle EBS interface development - quick generation of JSON format data
Network security -- intrusion detection of emergency response
A slide with two tables will help you quickly understand the target detection