olderth的博客

原创 使用MMDetection测试模型时，发生data[‘category_id‘] = self.cat_ids[label]的报错问题的解决方案

我自己的数据类别只有一个，在训练时并没报错，但在测试的时候报错了。我在网上搜索解决方案时大多数回答都是修改mmdetection/mmdet/datasets/coco.py文件中的类别数量。我自己也尝试了但没有解决，还是报这个错误。然后之前在使用官方文档安装mmdet时，官方有两种方案：而我在配置环境时，使用了mim install mmdet安装mmdetection，但我的代码也是在官方的源码上跑的，我每次修改的也是mmdetection中的文件。然后我就想是不是测试时导入的是第三方的mmdet。