最近mmdetection更新了2.0版本,更新了一点api。写一下自己的笔记吧
官方给出的测试demo是使用faster-rcnn模型,而我在使用别的模型时,尝试想画出mask的时候一直报错,如下:
Traceback (most recent call last):
File "demo.py", line 21, in <module>
show_result_pyplot(model, img, result, score_thr=0.3)
File "/mmdetection/mmdet/apis/inference.py", line 146, in show_result_pyplot
img = model.show_result(img, result, score_thr=score_thr, show=False)
File "/mmdetection/mmdet/models/detectors/base.py", line 211, in show_result
mask = maskUtils.decode(segms[i]).astype(np.bool)