一、导读

日常我们在艺术创作和产品设计中,需要多张风格相似的图片:

1. 艺术创作。根据一张原始图像,一键生成众多相似又不完全相同的图像,达到多样化的效果

2. 产品/平面设计。为产品生成相似的、多样的、不同分辨率的效果图,提供更多选择,节约时间和成本

魔搭社区近期上线了相似图像生成模型和体验创空间

输入示例:

输出示例:

 

 

 

 

二、模型下载&使用

相似图像生成模型现已在ModelScope社区开源,包括:

模型链接:https://modelscope.cn/models/damo/cv_image_variation_sd

相似图像生成创空间体验链接:https://modelscope.cn/studios/damo/image_variation

 

社区支持直接下载模型的repo:

from modelscope.hub.snapshot_download import snapshot_download

model_dir = snapshot_download('damo/cv_image_variation_sd', 'v1.1.0')

三、模型推理

依赖项:

pip install git+https://github.com/lllcho/image_variation.git

推理代码:

from modelscope.pipelines import pipeline
from modelscope.outputs import OutputKeys
from PIL import Image
from image_variation import modelscope_warpper

model = 'damo/cv_image_variation_sd'
pipe = pipeline('image_variation_task', model=model, device='gpu',auto_collate=False,model_revision='v1.1.0')
out=pipe('https://vision-poster.oss-cn-shanghai.aliyuncs.com/lllcho.lc/data/test_data/sunset-landscape-sky-colorful-preview.jpg')
imgs=out[OutputKeys.OUTPUT_IMGS]
imgs[0].save(f'result.jpg')

显存占用:

体验链接:https://modelscope.cn/studios/damo/image_variation/summary 

 

Logo

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!

更多推荐