一、导读
日常我们在艺术创作和产品设计中,需要多张风格相似的图片:
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