日常我们在艺术创作和产品设计中,需要多张风格相似的图片:
1. 艺术创作。根据一张原始图像,一键生成众多相似又不完全相同的图像,达到多样化的效果
2. 产品/平面设计。为产品生成相似的、多样的、不同分辨率的效果图,提供更多选择,节约时间和成本
魔搭社区近期上线了相似图像生成模型和体验创空间
输入示例:
输出示例:
相似图像生成模型现已在ModelScope社区开源,包括:
模型链接: https://modelscope.cn/models/damo/cv_image_variation_sd
相似图像生成创空间体验链接: https://modelscope.cn/studios/damo/image_variation
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
所有评论(0)