filaPro / oneformer3d

[CVPR2024] OneFormer3D: One Transformer for Unified Point Cloud Segmentation
Other
348 stars 32 forks source link

Using oneformer3d for semantic segmentation only #59

Closed boxiao-wv closed 5 months ago

boxiao-wv commented 5 months ago

I wonder whether there is a configuration template for purely semantic segmentation only? I understand that the current template outputs instance, semantic and panoptic segmentation results. Is there an easy way to disable the instance and panoptic segmentation parts?

Thank you very much!

oneformer3d-contributor commented 5 months ago

I think the easiest way is to set instance loss wieght to 0. Otherwise you need to modify the decoder (remove instance queries) and loss (remove instance loss).