当前位置:网站首页>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
边栏推荐
- Network security -- intrusion detection of emergency response
- Message queue fnd in Oracle EBS_ msg_ pub、fnd_ Application of message in pl/sql
- Jordan decomposition example of matrix
- Play online games with mame32k
- 【Ranking】Pre-trained Language Model based Ranking in Baidu Search
- sparksql数据倾斜那些事儿
- 机器学习理论学习:感知机
- Two dimensional array de duplication in PHP
- MySQL组合索引加不加ID
- 常见的机器学习相关评价指标
猜你喜欢
随机推荐
[paper introduction] r-drop: regulated dropout for neural networks
SSM student achievement information management system
CRP implementation methodology
Get the uppercase initials of Chinese Pinyin in PHP
【MEDICAL】Attend to Medical Ontologies: Content Selection for Clinical Abstractive Summarization
《Handwritten Mathematical Expression Recognition with Bidirectionally Trained Transformer》论文翻译
使用Matlab实现:Jacobi、Gauss-Seidel迭代
Find in laravel8_ in_ Usage of set and upsert
離線數倉和bi開發的實踐和思考
SSM personnel management system
腾讯机试题
[tricks] whiteningbert: an easy unsupervised sentence embedding approach
SSM二手交易网站
Module not found: Error: Can't resolve './$$_gendir/app/app.module.ngfactory'
view的绘制机制(二)
One field in thinkphp5 corresponds to multiple fuzzy queries
Oracle 11g sysaux table space full processing and the difference between move and shrink
传统目标检测笔记1__ Viola Jones
MySQL无order by的排序规则因素
Ding Dong, here comes the redis om object mapping framework