How to get Instance segmentation mask and class info (stored in a json or txt like file) for solov2 inference？

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.

Other

1.69k stars 307 forks source link

How to get Instance segmentation mask and class info (stored in a json or txt like file) for solov2 inference？ #182

Open yhe-rs opened 3 years ago

yhe-rs commented 3 years ago

Thanks for the amazing works of SOLOv2, it is the future of instance seg. I have one question during image inference using demo.py. The demo.py's output is a bunch of images showing instances on them. Is there a way to also output their associated info, (for instance a json or txt file), storing the class and coordinates of each mask?

Thanks in advance