search

hkchengrex / Cutie

[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation
https://hkchengrex.com/Cutie/
MIT License
579 stars 60 forks source link

Question: How to change dilation radius / padding of mask? #50

Closed realpixelbard closed 3 months ago

realpixelbard commented 4 months ago

Hi Rex and team, thank you for making this useful tool. 🙏

May I know how do we change the dilation radius (also known as padding width) of the mask?

hkchengrex commented 4 months ago

Dilation should be done in post-processing -- for ProPainter, you can use the --mask_dilation argument when running ProPainter. In general, you can use OpenCV's cv2.dilate.