简介
Lama Cleaner,基于 SOTA AI 模型构建,可用于快速去除图像中各种物品、人物、字体、水印等对象,并支持老照片修复、文本替换图像内容等。
项目免费开源,支持自托管。如果担心数据隐私安全的话,可以把项目托管到自己的服务器上。
安装
因国内网络,以及GPU的限制,建议直接使用官方提供的体验地址,地址在文末
pip方式安装
python版本:python 3.7 ~ 3.10
pip install lama-cleaner
# 通过命令启动 lama-cleaner 服务器,首次启动会下载模型
lama-cleaner --model=lama --device=cuda --port=8080
之后打开浏览器访问:http://localhost:8080
windows一键安装包
收费5美元,不用也罢
docker安装
CPU模式(不建议)
docker run -p 8080:8080 \
-v /path/to/cache_dir:/root/.cache/ \
--rm cwq1913/lama-cleaner:cpu-0.33.0 \
lama-cleaner --device=cpu --port=8080 --host=0.0.0.0
GPU模式
环境要求:
- cuda: 11.7
- pytorch: 1.13.1
- minimum nvidia driver: 515.43.04+
- xformers: 0.0.16rc425
docker run --gpus all -p 8080:8080 \
-v /path/to/cache_dir:/root/.cache/ \
--rm cwq1913/lama-cleaner:gpu-0.33.0 \
lama-cleaner --device=cuda --port=8080 --host=0.0.0.0
使用
上传图片,画笔涂抹需要处理的位置,随后自动开始处理,处理过程中屏幕会不断闪烁,处理完成还可以下载。
亲测效果还是挺惊艳的,不仅想到一个应用场景,视频水印去除功能。后期有机会使用该方案的接口处理视频水印看看有无可行性。
相关链接
github:https://github.com/Sanster/lama-cleaner
在线体验地址:https://huggingface.co/spaces/Sanster/Lama-Cleaner-lama