Small holes in the masks

[JianpingCAI]

Hi,

When I try on some images, there are small holes in some results, may I know how to reduce/remove these holes? is it a correct behavior?

Thanks,
Jianping

[lzhuwen]

Hi Jianping,

Thanks for your interests. One direct solution is to add more clicks if the holes are not very small; the other solution is to apply dilation and erosion to remove the very small holes. The pre-trained model is trained on the SBD dataset, you may also want to retrain it with your specific data.

Best,

Zhuwen

[JianpingCAI]

Hi Zhuwen,

Thanks a lot for your reply, and for updating the train code.
So the small holes issue is expected, right?

Sincerely,
Jianping