search

NVlabs / CenterPose

Single-Stage Keypoint-based Category-level Object Pose Estimation from an RGB Image (ICRA 2022)
Other
285 stars 34 forks source link

Way to generate category labels for CenterPose #20

Open ArghyaChatterjee opened 1 year ago

ArghyaChatterjee commented 1 year ago

Hello,

Thanks for your work. Is there a way to generate category labels in CenterPose as you do for DOPE during instance segmentation ? As there can be multiple categories, it's confusing if the network is not generating any labels for each category.

Thanks in advance.