inference_demo.inpynb D2Det-instance-res101.pth error

[zgq91]

Hi:
感谢开源，当我使用inference_demo.ipynb代码是 checkpoint 的模型D2Det-instance-res101.pth 会报以下的错误"d2det\mmdet\apis\inference.py", line 169, in show_result
img[mask] = img[mask] * 0.5 + color_mask * 0.5
IndexError: boolean index did not match indexed array along dimension 2; dimension is 3 but corresponding boolean dimension is 19"

[JialeCao001]

@zgq91 Hi. The problem has been fixed. Please update the functions in mmdet/apis/inference.py.

D2Det/mmdet/apis/inference.py

Line 119 in ca26d3f

def show_result(img,

D2Det/mmdet/apis/inference.py

Line 188 in ca26d3f

def show_result_pyplot(img,

You can also use our newly provided file to show or save detection results of a single image: demo/D2Det_demo.py. Please refer to README for the usage.

[zgq91]

谢谢