simple-lama-inpainting是LaMa的一个简便版本,已开发pypi packpage 安装包。https://github.com/advimman/lamaicon-default.png?t=N7T8https://github.com/advimman/lama

 simple-lama-inpainting 源码:GitHub - enesmsahin/simple-lama-inpaintingContribute to enesmsahin/simple-lama-inpainting development by creating an account on GitHub.icon-default.png?t=N7T8https://github.com/enesmsahin/simple-lama-inpainting/tree/main simple-lama-inpainting 安装包

simple-lama-inpainting · PyPIicon-default.png?t=N7T8https://pypi.org/project/simple-lama-inpainting/

安装命令

pip install simple-lama-inpainting

如果安装出现以下错误

则更换python3.9 以上版本,亲测可行。

使用:

CLI

simple_lama <path_to_input_image> <path_to_mask_image> <path_to_output_image>

代码中使用simple-lama-inpainting​​​​​​​模块,示例如下

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)

result = simple_lama(image, mask)
result.save("inpainted.png")

输入格式:

  • 输入图像可以是 np.ndarray(NumPy 数组)或 PIL.Image.Image(Pillow 库中的图像对象)。
  • 输入图像应为3通道图像,即彩色图像。
  • 另外,还需要一个单通道的二进制掩码图像,其中像素值为255的部分将被修复(inpainting)。

输出格式:

  • 输出图像的格式为 PIL.Image.Image

简而言之,这个任务的目标是通过提供的二进制掩码图像,对输入的3通道彩色图像进行修复,输出修复后的图像。

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐