mmdetection 训练COCO出现AttributeError ‘COCO‘ object has no attribute ‘get_cat_ids‘

最新推荐文章于 2024-05-12 11:18:11 发布

NH558J0

最新推荐文章于 2024-05-12 11:18:11 发布

阅读量3k

点赞数

CC 4.0 BY-SA版权

分类专栏： mmdetection 文章标签： mmdetection coco get_cat_ids

本文链接：https://blog.youkuaiyun.com/happyday_d/article/details/107388511

mmdetection 专栏收录该内容

1 篇文章

订阅专栏

本文详细解析了在使用mmdetection进行COCO数据集训练时遇到的AttributeError错误，即‘COCO‘对象不存在‘get_cat_ids‘属性的问题。深入探讨了错误的根源在于mmdetection与pycocotools之间的接口不兼容，并提供了两种解决方案：一是通过git回滚coco.py文件到先前版本并重新安装兼容的cocoAPI；二是直接修改相关函数名称以实现快速修复。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

mmdetection 训练COCO出现AttributeError: ‘COCO‘ object has no attribute ‘get_cat_ids‘

原因

mmdetection用到了pycocotools，本来接口是pycocotools.coco.getCatIds，但是mmdetection又在外包了一层，变成了get_cat_ids，而pycocotools.coco又没有这个成员函数，导致出错。

解决方法

方法1，参考github issue：

revert the coco.py file to the previous version via git checkout 206107 -- mmdet/datasets/coco.py
reinstall the mmdet compatible version of cocoAPI by pip install -U "git+https://github.com/open-mmlab/cocoapi.git#subdirectory=pycocotools"

方法2，手动修改（未连网情况下适用）

get_cat_ids ---> getCatIds
get_img_ids ---> getImgIds
cat_names   ---> catNms
load_imgs   ---> loadImgs
load_anns   ---> loadAnns
get_anno_ids(img_ids=) ---> getAnnIds(imgIds=)