search

NVlabs / ODISE

Official PyTorch implementation of ODISE: Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models [CVPR 2023 Highlight]
https://arxiv.org/abs/2303.04803
Other
845 stars 45 forks source link

predictions['sem_seg'] does not contain the background class #20

Open YuhuiWang99 opened 1 year ago

YuhuiWang99 commented 1 year ago

The shape of predictions['sem_seg'] is [133, H, W] for coco, does it contain the background class, then how to get the correct semantic segmentation output？