Stable Diffusion中的图像修改尺寸和编辑工具实现

图生图

在图生图中,使用Stablediffusion的pipeline来构建Webui应用,如果直接把参数传给pipeline是无效的。 __call__参数中虽然有单独的weightheight 这两个参数,但最终图像尺寸实际上是由传入图片的尺寸决定的,因此需要定义函数来做原始图片的尺寸变换,达到用参数来控制输出图像尺寸的目的,这可以通过resize调整图片尺寸大小来实现。

python 复制代码
# 图片宽度和高度按照比例调整
def fn_resize_ratio(original_image, ratio):
    
    original_width, original_height = original_image.size
    new_width, new_height  = int(ratio * original_width), int(ratio * original_height) 
    
    while not new_width%8==0:
        new_width += 1
    while not new_height%8==0:
        new_height += 1

    # Resize the image
    resized_image = original_image.resize((new_width, new_height))

    return resized_image, new_width, new_height
python 复制代码
# 图片宽度和高度按照不同尺寸调整
def fn_resize_fixed(original_image, new_width, new_height):
    resized_image = original_image.resize((new_width, new_height))
    return resized_image

上传一张初始图片 图片宽度和高度按照比例调整,缩放到1.2倍,高度和高度同步变成1232。

图片宽度和高度按照不同尺寸调整,可以看到图片被「压扁」了。

图像编辑

Gradio 4.0以上的版本提供了ImageEditor 的组件,可以很方便实现图片的裁剪,涂改等,非常适合图像编辑目的。 ImageEditor 是一个图像组件,可用于上传和编辑图像(作为输入)或显示图像(作为输出)。 但是上传后的图像不是一个PIL.Image对象,而是变成一个字典。包含background、layers 和composite。background是上传的原始图像,layers表述的是当前层的操作的痕迹,如裁剪或者涂改。

python 复制代码
# 提取图像编辑器的涂鸦区域和原图
def save_mask(inputs):
    layers = inputs['layers']
   
    # layers[0]表示第一张mask
    mask = layers[0]
    new_image = Image.new('RGBA', mask.size, color='white').paste(mask, mask=mask).convert('RGB')
    return ImageOps.invert(new_image), inputs['background']

图像编辑器 图像做了涂鸦,右侧输出为涂鸦的区域 图像编辑器的层数,默认是1。如果需要多个涂鸦层,可以点击「+」号增加。不同的layer之间不会互相影响。

相关推荐
FuckTheWindows12 分钟前
Transformer中Decoder模块是如何预测下一个字符的算法
深度学习·算法·transformer
程序员阿瓜15 分钟前
基于Spring AI开发本地Jenkins MCP Server服务
人工智能
机器之心24 分钟前
3D领域DeepSeek「源神」启动!国产明星创业公司,一口气开源八大项目
人工智能
量子位26 分钟前
Manus 欲以 37 亿估值硅谷寻融资!发布仅三周,线下用户聚会场场满员
人工智能
量子位26 分钟前
AI 大模型看手相!图片视频加持深度思考,阿里 QVQ-Max“神了神了”
人工智能·aigc
敲代码中30 分钟前
位运算[找出唯一成对的数]
java·算法·排序算法
郝YH是人间理想34 分钟前
OpenCV基础——图像滤波和形态学操作
图像处理·人工智能·opencv
trust Tomorrow1 小时前
每日一题-力扣-2712: 使所有字符相等的最小成本 0327
python·算法·leetcode
京东零售技术1 小时前
京东广告创意:高质量创意生成和千人千面的创意推荐 | 京东零售技术实践
人工智能
羊仔AI探索1 小时前
AI的未来在手机里!
人工智能·ai·智能手机·aigc