advimman / lama

🦙 LaMa Image Inpainting, Resolution-robust Large Mask Inpainting with Fourier Convolutions, WACV 2022
https://advimman.github.io/lama-project/
Apache License 2.0
8.14k stars 864 forks source link

ignore pixels #335

Open MRamazan opened 2 months ago

MRamazan commented 2 months ago

Is it possible for the inpainting model to ignore certain parts of the image when making predictions?For example, I have a picture like this. While inpainting certain parts of this image, I want black pixels not to be used when making predictions. Is this possible? image2 this is LaMa inpainting model output: image2_mask001