当前位置:网站首页>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
边栏推荐
- Two table Association of pyspark in idea2020 (field names are the same)
- SSM student achievement information management system
- ssm+mysql实现进销存系统
- Explanation of suffix of Oracle EBS standard table
- Spark SQL task performance optimization (basic)
- Oracle EBS database monitoring -zabbix+zabbix-agent2+orabbix
- SSM supermarket order management system
- Agile development of software development pattern (scrum)
- Oracle EBS DataGuard setup
- Network security -- intrusion detection of emergency response
猜你喜欢
随机推荐
【Torch】解决tensor参数有梯度,weight不更新的若干思路
【信息检索导论】第六章 词项权重及向量空间模型
Open failed: enoent (no such file or directory) / (operation not permitted)
【MEDICAL】Attend to Medical Ontologies: Content Selection for Clinical Abstractive Summarization
【Torch】最简洁logging使用指南
Pratique et réflexion sur l'entrepôt de données hors ligne et le développement Bi
【BERT,GPT+KG调研】Pretrain model融合knowledge的论文集锦
ARP attack
Ding Dong, here comes the redis om object mapping framework
常见CNN网络创新点
Conda 创建,复制,分享虚拟环境
Using MATLAB to realize: Jacobi, Gauss Seidel iteration
Huawei machine test questions
ssm+mysql实现进销存系统
Transform the tree structure into array in PHP (flatten the tree structure and keep the sorting of upper and lower levels)
JSP智能小区物业管理系统
Jordan decomposition example of matrix
Find in laravel8_ in_ Usage of set and upsert
使用Matlab实现:弦截法、二分法、CG法,求零点、解方程
view的绘制机制(三)









