mmedetection 训练数据集问题

最新推荐文章于 2023-10-07 09:26:02 发布

空心22

最新推荐文章于 2023-10-07 09:26:02 发布

阅读量1.3k

点赞数

CC 4.0 BY-SA版权

分类专栏： solov2训练文章标签：深度学习 pytorch 人工智能

本文链接：https://blog.youkuaiyun.com/qq_52103624/article/details/125262281

solov2训练专栏收录该内容

1 篇文章

订阅专栏

博客讨论了在使用PyCOCOTools时遇到的TypeError，原因是将圆心标注误识别为bbox。问题根源在于标注数据格式不正确，圆心数据被当作bbox处理。解决方案是检查并修正标注数据，确保它们符合polygon格式。此外，还提供了GitHub上的相关issue链接作为参考。

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

报错：

 File "pycocotools/_mask.pyx", line 294, in pycocotools._mask.frPyObjects
TypeError: Argument 'bb' has incorrect type (expected numpy.ndarray, got list)

这个问题很阴险，个人理解是训练过程中如果遇到segmentation中数据个数恰好是四个，如[x,ｙ,ｘ,ｙ]，那么就会把它当做bbox,因为bbox中数据恰好是四个（左上和右下）。。。。网上有人说时版本问题，但是本人的pycocotools是2.0.4，好像不是版本太低的原因，暂且认为是一个没有解决的bug(因为并没有仔细看mask代码中是怎么挑出bbox数据的，难道是比对个数？？？）

找到对应的图片后，发现是因为神经网络训练的标注类型只能是polygon,而图片里用了labelme中的圆形工具来偷懒，导致存的数据只有圆心和圆上一个点的位置，当然就只有四个数。

比较有用的解决方案是

TypeError: Argument 'bb' has incorrect type (expected numpy.ndarray, got list) · Issue #139 · cocodataset/cocoapi · GitHubhttps://github.com/cocodataset/cocoapi/issues/139按照人家说的直接找到删了就不报错了。