mmedetection 训练数据集问题

报错:

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

这个问题很阴险,个人理解是训练过程中如果遇到segmentation中数据个数恰好是四个,如[x,y,x,y],那么就会把它当做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 · GitHubicon-default.png?t=M4ADhttps://github.com/cocodataset/cocoapi/issues/139按照人家说的直接找到删了就不报错了。

 
 

你可能感兴趣的:(solov2训练,深度学习,pytorch,人工智能)