lama-cleaner中的Paint-by-Example使用

Lama Cleaner是一个免费的、开源的、完全自托管的修复工具,里面提供了很多最前沿的AIGC模型。可以使用它从图片中删除任何不需要的物体、缺陷、人物,或删除和替换图片上的任何内容。本文章详细介绍了该工具的所有功能,并体验了下每个功能的实际效果。

一 安装

# pip直接安装
pip install lama-cleaner

本文安装时版本更新到1.2.5

二 Paint By Example图像物体插入

github:https://github.com/Fantasy-Studio/Paint-by-Example

paper:Paint by Example: Exemplar-based Image Editing with Diffusion Models
因为需要的数据集无法构建,提出替代方案,用一张图的检测框作为mask,框里面的图就是参考图。把参考图里面的目标还原回原始图里。把还原回来的图和真实的原图做损失计算。
为了保证该方法不偷懒,直接学复制粘贴操作。分别提出下面两点创新:
利用预训练的SD作初始化,把参考图提取特征后只保留CLS的token,再通过Cross Attention的方式,加上一些额外的全连接层到扩散过程中去解码特征。避免图像特征给多了让神经网络偷懒。
参考图做数据增强,(反转,平移等),检测框的mask形状做更多的随机调整,模拟真实性。

下面的操作就是先是选定两张图片,一张为要放置物体的原图(该原图要涂要添加物体的区域),另外一张为物体图像(将图片中的物体放在第一张图片的指定区域)
在这里插入图片描述
在这里插入图片描述
自动下载模型的指令

lama-cleaner --model=paint_by_example --device=cuda --port=8080

手动下载模型的指令
把https://huggingface.co/Fantasy-Studio/Paint-by-Example/tree/main的模型都下载,保存在某个目录中

lama-cleaner --model=paint_by_example --device=cuda --port=8080 --sd-local-model-path=/home/***/Fantasy-Studio/Paint-by-Example

备注:绝对路径和相对路径

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值