rasmushaugaard / surfemb

SurfEmb (CVPR 2022)
https://surfemb.github.io/
MIT License
77 stars 17 forks source link

question about bbox and mask #22

Closed smoothumut closed 2 years ago

smoothumut commented 2 years ago

Thanks again for this great work. I am still trying to make it work for our case. In order to run the model with our custom dataset, should we need to have bbox and mask in our bop formatted custom dataset? we can provide bbox_visible but not bbox. and we can provide mask_visible but not mask? how important are they? would be bbox_visible and mask_visible enough? if we need them for this model, do you have any idea how to generate that ? thanks thanks in advance

rasmushaugaard commented 2 years ago

The full mask is found by rendering the object in the ground truth pose during training. You don't need to provide bbox and mask, but you would probably need to remove the "bbox_obj" from surfemb/data/instance.py.

smoothumut commented 2 years ago

thanks a lot for the answer. Deeply appreciated