Python图像叠加问题：毛衣部件合成蒙版异常排查

问题概述 #

开发图像部件合成项目，需将毛衣前片、左右袖子图像合成为完整毛衣效果，目标是构建生成合成图像的API。目前已用rembg成功移除图像背景，通过PIL创建蒙版并尝试叠加图像，但合成结果与Photoshop制作的目标效果差异明显，怀疑蒙版设置存在问题但无法定位原因。

现有代码 #

from rembg import remove
from PIL import Image, ImageDraw, ImageFilter
import cv2
import numpy as np

background = Image.open('Front.png')
foreground = Image.open('SleeveR.png')
foreground1 = Image.open('SleeveL.png')

width, height = foreground.size
mask_im = Image.new("L", foreground.size, 0)
draw = ImageDraw.Draw(mask_im)
draw.rectangle((130, 70, width/2, height), fill=255)
mask_im = mask_im.rotate(-20, resample=Image.NEAREST, expand=False)
mask_im.save('mask_circle.jpg', quality=95)

background.paste(foreground, (-120, 0), mask_im)
background.save('result.png', quality=100)

图像资源 #

• 毛衣前片
• 左袖图像
• 右袖图像

问题分析与修复建议 #

1. 蒙版旋转与边缘处理问题 #

• 用Image.NEAREST重采样旋转蒙版会产生锯齿边缘，导致袖子叠加后生硬，和PS平滑效果差距大。建议改用Image.BILINEAR或Image.BICUBIC，同时开启expand=True避免蒙版内容被裁切：

  mask_im = mask_im.rotate(-20, resample=Image.BILINEAR, expand=True)
  

• 硬边缘矩形蒙版无法实现PS中的自然融合效果，给蒙版添加高斯模糊实现羽化过渡：

  mask_im = mask_im.filter(ImageFilter.GaussianBlur(radius=3))
  

2. 前景与蒙版的同步匹配问题 #

• 旋转蒙版后尺寸变化，原粘贴位置会导致蒙版与前景错位，需同步旋转前景图像并调整偏移：

  # 同步旋转前景和蒙版
  foreground_rotated = foreground.rotate(-20, resample=Image.BILINEAR, expand=True)
  mask_im_rotated = mask_im.rotate(-20, resample=Image.BILINEAR, expand=True)
  # 调整粘贴位置
  paste_x = -120 + (foreground_rotated.width - foreground.width) // 2
  paste_y = 0 + (foreground_rotated.height - foreground.height) // 2
  background.paste(foreground_rotated, (paste_x, paste_y), mask_im_rotated)
  

3. 蒙版存储格式问题 #

• JPG是有损压缩格式，会丢失蒙版灰度细节，建议保存为PNG格式：

  mask_im.save('mask_circle.png')
  

4. 左右袖子对称处理 #

• 当前仅处理右袖，左袖需创建对称蒙版，调整旋转角度为正20度左右，并将粘贴位置移至左侧对应区域，保证效果对称。

优化后的完整代码示例 #

from rembg import remove
from PIL import Image, ImageDraw, ImageFilter

# 加载图像
background = Image.open('Front.png')
right_sleeve = Image.open('SleeveR.png')
left_sleeve = Image.open('SleeveL.png')

def process_sleeve(sleeve_img, rotate_angle, paste_offset, mask_rect):
    # 创建蒙版
    mask = Image.new("L", sleeve_img.size, 0)
    draw = ImageDraw.Draw(mask)
    draw.rectangle(mask_rect, fill=255)
    # 羽化蒙版边缘
    mask = mask.filter(ImageFilter.GaussianBlur(radius=4))
    # 同步旋转袖子和蒙版
    sleeve_rotated = sleeve_img.rotate(rotate_angle, resample=Image.BILINEAR, expand=True)
    mask_rotated = mask.rotate(rotate_angle, resample=Image.BILINEAR, expand=True)
    # 调整粘贴位置
    paste_x = paste_offset[0] + (sleeve_rotated.width - sleeve_img.width) // 2
    paste_y = paste_offset[1] + (sleeve_rotated.height - sleeve_img.height) // 2
    return sleeve_rotated, mask_rotated, (paste_x, paste_y)

# 处理右袖
right_rect = (130, 70, right_sleeve.width/2, right_sleeve.height)
right_sleeve_rot, right_mask_rot, right_pos = process_sleeve(right_sleeve, -20, (-120, 0), right_rect)
background.paste(right_sleeve_rot, right_pos, right_mask_rot)

# 处理左袖
left_rect = (right_sleeve.width/2, 70, right_sleeve.width - 130, right_sleeve.height)
left_sleeve_rot, left_mask_rot, left_pos = process_sleeve(left_sleeve, 20, (background.width - left_sleeve.width + 120, 0), left_rect)
background.paste(left_sleeve_rot, left_pos, left_mask_rot)

# 保存结果
background.save('final_sweater.png', quality=100)

内容的提问来源于stack exchange，提问作者mrconcerned