Simple pip package for LaMa[1] inpainting.
pip install simple-lama-inpainting
simple_lama <path_to_input_image> <path_to_mask_image> <path_to_output_image>
Input formats: np.ndarray or PIL.Image.Image. (3 channel input image & 1 channel binary mask image where pixels with 255 will be inpainted).
Output format: PIL.Image.Image
from simple_lama_inpainting import SimpleLama
from PIL import Image
simple_lama = SimpleLama()
img_path = "image.png"
mask_path = "mask.png"
image = Image.open(img_path)
mask = Image.open(mask_path).convert('L')
result = simple_lama(image, mask)
result.save("inpainted.png")[1] Suvorov, R., Logacheva, E., Mashikhin, A., Remizova, A., Ashukha, A., Silvestrov, A., Kong, N., Goka, H., Park, K., & Lempitsky, V. (2021). Resolution-robust Large Mask Inpainting with Fourier Convolutions. arXiv preprint arXiv:2109.07161.
[2] https://github.com/saic-mdal/lama
[3] https://github.com/Sanster/lama-cleaner