search

SHI-Labs / OneFormer

OneFormer: One Transformer to Rule Universal Image Segmentation, arxiv 2022 / CVPR 2023
https://praeclarumjj3.github.io/oneformer
MIT License
1.41k stars 128 forks source link

Bounding box for each instance #41

Closed rose-jinyang closed 1 year ago

rose-jinyang commented 1 year ago

Hello How are you? Thanks for contributing to this project. I found that the demo script does NOT output bounding boxes for each instance in panoptic/instance segmentation. Could u guide me how to get the bounding boxes?

praeclarumjj3 commented 1 year ago

Hi @rose-jinyang, please set DETECTION_ON as True in the config to visualize bounding boxes.

https://github.com/SHI-Labs/OneFormer/blob/761189909f392a110a4ead574d85ed3a17fbc8a7/configs/coco/oneformer_R50_bs16_50ep.yaml#L54

rose-jinyang commented 1 year ago

It works now. Thanks