SOLOv2 get segmentation mask and classes

[williamcfrancis]

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.