图像示例替换

我要开发同款
匿名用户2024年07月31日
62阅读

技术信息

开源地址
https://modelscope.cn/models/iic/cv_stable-diffusion_paint-by-example
授权协议
Apache License 2.0

作品详情

Pait by Example 图像示例替换

本模型选自Pait by Example, 根据示例图片对原始图片的mask区域进行自适应地替换。

模型描述

Pait by Example 是基于stable diffusio 模型的一个图像编辑算法,根据示例图片对原始图片的mask区域进行自适应地替换。

期望模型使用方式以及适用范围

本模型适用范围为室外自然场景;

如何使用

在ModelScope框架上,提供输入图片,即可通过简单的Pipelie调用来使用。

环境安装

安装好基础modelscope环境后,安装pait-ldm:

pip istall pait-ldm -f https://modelscope.oss-c-beijig.aliyucs.com/releases/repo.html

代码范例

  • 推理(仅支持GPU):
from PIL import Image
import cv2
from modelscope.outputs import OutputKeys
from modelscope.pipelies import pipelie
from modelscope.utils.costat import Tasks

iput_locatio = 'https://modelscope.oss-c-beijig.aliyucs.com/test/images/image_pait_by_example/image/example_1.pg'
iput_mask_locatio = 'https://modelscope.oss-c-beijig.aliyucs.com/test/images/image_pait_by_example/mask/example_1.pg'
referece_locatio = 'https://modelscope.oss-c-beijig.aliyucs.com/test/images/image_pait_by_example/referece/example_1.jpg'
iput = {
        'img':iput_locatio,
        'mask':iput_mask_locatio,
        'referece':referece_locatio,
}

paitbyexample = pipelie(Tasks.image_paitbyexample, model='damo/cv_stable-diffusio_pait-by-example')
result = paitbyexample(iput)
vis_img = result[OutputKeys.OUTPUT_IMG]
cv2.imwrite("result.pg", vis_img)

模型局限性以及可能的偏差

  • 人脸图片暂不支持
  • 当前版本在pytho 3.8环境测试通过,其他环境下可用性待测试
  • 当前版本fie-tue在cpu和单机单gpu环境测试通过,单机多gpu等其他环境待测试

训练数据介绍

引用

如果你觉得这个该模型对有所帮助,请考虑引用下面的相关的论文:

@article{yag2022pait,
  title={Pait by Example: Exemplar-based Image Editig with Diffusio Models},
  author={Bixi Yag ad Shuyag Gu ad Bo Zhag ad Tig Zhag ad Xueji Che ad Xiaoya Su ad Dog Che ad Fag We},
  joural={arXiv preprit arXiv:2211.13227},
  year={2022}
}

功能介绍

Paint by Example 图像示例替换 本模型选自Paint by Example, 根据示例图片对原始图片的mask区域进行自适应地替换。 模型描述 Paint by Example 是基

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论