How to determine what each ID corresponds to in terms of objects?

lkeab / gaussian-grouping

[ECCV'2024] Gaussian Grouping for open-world Anything reconstruction, segmentation and editing.

https://arxiv.org/abs/2312.00732

Apache License 2.0

500 stars 37 forks source link

How to determine what each ID corresponds to in terms of objects? #13

Open baiyunshu opened 5 months ago

baiyunshu commented 5 months ago

i want to change the id but how can i know what object does each id represent?

ymq2017 commented 5 months ago

Hi, we get this object id from the training label in data/bear/object_mask folder. You can check the pixel value of the label map for corresponding object.

Another way for text-prompt, since SAM masks are class-agnostic, we can use the mask output of grounded-sam to match our label to give semantic information.