Python开源工具库使用之图片去水印IOPaint

前言

最近发现一款去除图片水印的开源软件 IOPaint，感觉很不错，分享一下。它是一款基于 SOTA AI 模型的软件，可以实现从图片中擦除任何不需要的物体、缺陷、人，还可以替换掉(通过stable diffusion)，并且还能够支持 CPU 运行(当然替换功能 stable diffusion 使用 CPU 太慢了，不建议使用），以往的深度学习软件只能使用 GPU，这款工具还可以支持 CPU 就很满足需求。并且这款工具还通过插件支持其它功能，如脸部修复、图片超分辨率、移除背景等

• github地址：https://github.com/Sanster/IOPaint

一、去除水印

1.1 安装部署

通过pip安装

pip install torch==2.1.2 torchvision==0.16.2
pip install iopaint

在安装完成后，通过如下命令启动，可以指定 model 路径

iopaint start --model=lama --model-dir=models --device=cpu --port=8080

执行命令后，会从 github上下载模型文件，如果下载太慢，可手动下载，并放在models\torch\hub\checkpoints下面

github模型文件：https://github.com/Sanster/models/releases/download/add_big_lama/big-lama.pt

1.2 实战

当部署完成后，可通过localhost:8080来访问web界面

二、面部修复

2.1 安装运行

按照下面命令安装运行

pip install facexlib
pip install tb-nightly -i https://mirrors.aliyun.com/pypi/simple
pip install gfpgan
iopaint start --model=lama --model-dir=models --device=cpu --port=