mmdetection v2.0训练自己的voc数据集

修改步骤

首先按照 链接进行修改mmdetection中对应文件代码,但是有可能会报下面这个错误:

AssertionError: The `num_classes` (3) in Shared2FCBBoxHead of MMDataParallel does not matches the length of `CLASSES` 20) in RepeatDataset

原因分析:

数据集的类别信息(我的数据集类别是3)仍是VOC类别(20类)。

解决办法:

在修改完 class_names.py 和 voc.py 之后一定要重新编译代码(运行python setup.py install),再进行训练。

参考:
https://blog.csdn.net/weixin_42146720/article/details/114543889

版权声明:本文为CSDN博主「ydwl_懒洋洋」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/ydwl_lyy/article/details/115479566

ydwl_懒洋洋

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

分享 | 物体检测和数据集

因为最近学习任务比较紧(但也不妨碍元旦摆烂三天),所以中间有几个实战Kaggle比赛就跳过了,等以后有时间再回头来看看。物体检测和数据集这一节花了有一天的时间,一直有一个bug困扰,后来改了代码把box

深度学习——PaddlePaddle目标识别昆虫项目(2)

摘要 本文主要介绍的是的有关于的百度飞桨的目标的检测的算法的实战案例。通过使用的百度飞桨来实现的对自己目标进行识别的。后面本人将继续深入原理来对的目标的检测的原理和算法和系统的来实现讲解。基于飞桨(PaddlePaddle&#x