How to show the instance segmentation result

[freebooterish]

Thanks for your hard work. Much appreciated.
I try to run the command:
"python ./demo/D2Det_demo.py ./configs/D2Det/D2Det_instance_r101_fpn_2x.py ./D2Det-instance-res101.pth ./demo/demo.jpg --out ./demo/aa.jpg"
But aa.jpg only displays the detection results, NOT the segmentation results, Any ideas?

Another question: If I train with custom dataset, using the config "D2Det/D2Det_instance_r101_fpn_2x.py", Does the result contain segmentation mask? Do I need to add "mask_head" items to the configuration file?

Thank you!

[JialeCao001]

@freebooterish Thanks for your interest. Currently, I am sorry that I have no time to help check the problem. Maybe you can try to debug it by checking the imshow function. If you train the model with D2Det/D2Det_instance_r101_fpn_2x.py, I thinkt it should contain the segmentation mask because the config file will set the MASK_ON be true.