simple-lama-inpainting 安装使用,集成到自己的代码里

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

simple-lama-inpainting 安装使用,集成到自己的代码里_第1张图片

 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/simple-lama-inpainting 安装使用,集成到自己的代码里_第2张图片

安装命令

pip install simple-lama-inpainting

如果安装出现以下错误

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

使用:

CLI

simple_lama   

代码中使用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通道彩色图像进行修复,输出修复后的图像。

 

你可能感兴趣的:(python,机器学习,python,人工智能,图像修复)