yolov3训练自己的数据集(MMDetection)

2023-11-02

用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
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

yolov3训练自己的数据集(MMDetection) 的相关文章

  • log4cplus源码分析

    1 引题 虽然从本科起就学的C 然后在工作的2年时间中也不断的在用C 写代码 虽然基本的语法和一些常用的库函数已经滚瓜烂熟 可是总觉得自己写的代码还不是很专业 特别是看到那些老外们写得代码 从设计 到编码风格 再到各种编程技法的使用有很多都
  • opencv-python图像处理 ---图像轮廓检测与代码实现

    一 轮廓检测 边缘检测虽然能够检测边缘 但是其边缘并不是连续的 图像轮廓用于检测一个整体 来用于后续进行其他处理 opencv提供了两个函数来完成这些操作 findContours 该函数可以查找轮廓 drawContours 绘制轮廓 轮
  • 用keil的时候没有安装对应MCU的库,以STM32F103F为例

    安装keil的时候没有用到这个芯片就没有安装对应的库 重新安装之后遇到的几个坑 打开keil显示没有这个型号 解决方法是安装对应的库 STM32F103F要安装Keil STM32F1xx DFP 2 4 1 pack 安装完库之后 点击O
  • pytorch部署新利器TorchServer

    pytorch的爬坑指南 排坑不易转载请注明出处 参考文档 https github com pytorch serve 由于本地下载pytorch 1 7有问题 刚又瞅一眼现在是需要1 6 所以采用了docker部署 上干货 1 dock
  • BS4爬取物价局房产备案价以及dataframe的操作来获取房价的信息分析

    因为最近要买房子 然后对房市做了一些调研 发现套路极多 卖房子的顾问目前基本都是一派胡言能忽悠就忽悠 所以基本他们的话是不能信的 一个楼盘一次开盘基本上都是200 300套房子 数据量虽然不大 但是其实看一下也很烦要一页一页的翻 如果是在纸
  • SQL11 获取所有员工当前的manager

    SQL11 获取所有员工当前的manager 一 问题描述 二 基本思路 三 完成代码 一 问题描述 二 基本思路 员工表包括普通员工和经理 员工表左连接部门经理表 然后where筛选非经理的员工即可 三 完成代码 select dept

随机推荐