WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.7k stars 306 forks source link

SOLOv2 get segmentation mask and classes #237

Open williamcfrancis opened 1 year ago

williamcfrancis commented 1 year ago

Hi,

Solov2 is a very powerful tool for instance segmentation. I am trying to experiment with its output but this implementation uses mmdetection APIs which output a bunch of images showing instances on them. Is there a way to get the actual output from the model, i.e. the class and segmentation masks? I want to pass an image to the model at inference and get the raw output from the model. Any help is greatly appreciated.