如何在Numpy图像阵列中引入透明区域？

Question

输入图像已加载：

target_img = spi.imread(target_img, mode="RGB").astype(float)/256
source_img = spi.imread(source_img, mode="RGB").astype(float)/256
target_mask = spi.imread(target_mask, mode="RGB").astype(float)
source_mask = spi.imread(source_mask, mode="RGB").astype(float)

设置了一些变量：

target_mask_list = []
source_mask_list = []
mask_list = []
i = 0
matched_piece = None

我使用它们的蒙版图像提取目标图像和源图像的蒙版区域：

for color in list(color_list):
    print(color)
    #Get Target Mask Regions
    target_color_mask = extract_mask(target_mask, color)
    masked_target = np.where(target_color_mask[...,None], target_img, 0)
    #imsave(output_name, image_mask)
    target_mask_list.append(masked_target)
    #Get Source Mask Regions
    source_color_mask = extract_mask(source_mask, color)
    masked_source = np.where(source_color_mask[...,None], source_img, 0)
    source_mask_list.append(masked_source)

当前，“非选定”区域全是黑色还是全白色，这取决于我在np.where使用0还是1 。 “未选中”区域的不透明度为零，因此我无法正确地对这些图像进行分层。

然后，我通过颜色传递函数运行蒙版：

    # Perform Color Transfer
    matched_mask = match_color(masked_target, masked_source, mode='pca', eps=1e-5)
    mask_list.append(matched_mask)
    if i == 0: 
        matched_piece = np.where(target_color_mask[...,None], matched_mask, 0)
    else: 
        matched_piece = matched_piece + np.where(target_color_mask[...,None], matched_mask, 0)
    i = i + 1  

结果保存的图像是全黑的：

masked = matched_piece
imsave(output_name, masked)

当我使用GIMP使target_img或source_img具有透明部分时，我可以通过颜色传递函数成功运行这些图像，以便在非透明区域之间传递颜色。 然后，我可以手动裁剪出先前的透明部分，以便获得彩色输出图像。

如何使我的代码更像GIMP手动步骤那样工作？

编辑：

extract_mask函数从以下位置返回一个布尔数组：

   def extract_mask(image, color_list):
       mask = None
       if color == 'green':
           mask = np.all(image == (0,255,0), axis=-1).astype(int)   
       elif color == 'black': 
           mask = np.all(image == (0,0,0), axis=-1).astype(int)
       elif color == 'white':
           mask = np.all(image == (255,255,255), axis=-1).astype(int)
       elif color == 'red':
           mask = np.all(image == (255,0,0), axis=-1).astype(int)
       elif color == 'blue':
           mask = np.all(image == (0,0,255), axis=-1).astype(int)
       elif color == 'yellow':
           mask = np.all(image == (255,255,0), axis=-1).astype(int)
       elif color == 'grey':
           mask = np.all(image == (128,128,128), axis=-1).astype(int)
       elif color == 'lightblue':
           mask = np.all(image == (0,255,255), axis=-1).astype(int)
       elif color == 'purple':
           mask = np.all(image == (255,0,255), axis=-1).astype(int)
       else: 
           print "Color not recognized"
       return mask