如何在Python中自动生成异常图像及对应黑白掩码

要自动生成符合要求的掩码（异常区域白色，其余黑色），可以在现有代码基础上添加掩码生成逻辑，核心是利用异常图像的透明通道（Alpha通道）定位异常区域，具体步骤和修改后的代码如下：

实现思路 #

• 先创建一张和原始图像尺寸完全一致的全黑单通道图像作为掩码基底
• 提取异常图像的Alpha通道，把非透明区域（Alpha值大于0的部分）转换为白色，得到仅保留异常区域的掩码层
• 将这个白色异常区域的掩码层，按照异常图像粘贴到原图的相同坐标位置，粘贴到全黑基底上
• 最后保存生成好的掩码图像

修改后的完整代码 #

from PIL import Image
import random

anomaly = Image.open("anomaly_transp.png")  # 带透明背景的异常图像
image_number = 1  # 要生成的图像数量
w, h = anomaly.size
offset_x, offset_y = 480 - w, 512 - h  # 避免粘贴区域超出原图的偏移量

for i in range(image_number):
    original = Image.open("Crop_120.png")  # 原始正常图像
    x1 = random.randint(0, offset_x)
    y1 = random.randint(0, offset_y)
    area = (x1, y1, x1 + w, y1 + h)
    
    # 生成带异常的输出图像（原有逻辑保留）
    original.paste(anomaly, area, anomaly)
    original.save(f"output_{i}_.png")
    original.close()
    
    # 生成对应掩码
    # 1. 创建全黑单通道掩码基底（模式'L'表示8位灰度）
    mask_img = Image.new('L', (480, 512), 0)
    # 2. 从异常图像提取Alpha通道，将非透明区域转为白色（255）
    anomaly_mask = anomaly.getchannel('A').point(lambda x: 255 if x > 0 else 0)
    # 3. 将白色异常区域粘贴到掩码的对应位置
    mask_img.paste(anomaly_mask, area)
    # 4. 保存掩码图像
    mask_img.save(f"mask_{i}_.png")

关键代码解释 #

• Image.new('L', (480, 512), 0)：创建尺寸为480x512的全黑灰度图像，作为掩码基底
• anomaly.getchannel('A')：提取异常图像的Alpha通道，Alpha值0表示完全透明，255表示完全不透明
• point(lambda x: 255 if x > 0 else 0)：把Alpha通道中所有非透明的像素（x>0）转为白色（255），透明部分保持黑色（0），得到异常区域的白色掩码层
• mask_img.paste(anomaly_mask, area)：将白色异常区域粘贴到掩码的对应位置，和异常图像在原图上的位置完全对齐

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