search

trqminh / aistron

Amodal Instance Segmentation Toolbox and Benchmark
Apache License 2.0
19 stars 1 forks source link

Can not extract "visible masks" #5

Closed binh0804 closed 1 month ago

binh0804 commented 2 months ago

I am training on my own dataset but when I run inference, only pred_masks property have value (but fortunately it still amodal masks). Other properties, pred_amodal_masks, visible_masks, occluding_masks just only [] -> empty polygons. Is there any secret to train on custom dataset?

trqminh commented 1 month ago

Hi @binh0804. Thank you for your interest in our work! What method did you use to train? I guess you're using MaskRCNN. If that is the case, maskrcnn won't output pred_amodal_masks, visible_masks, occluding_masks