修改步骤
首先按照 链接进行修改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
暂无评论