svjack commited on
Commit
3d2f8c8
·
verified ·
1 Parent(s): b6938c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -50,16 +50,17 @@ def process_image(img):
50
 
51
  # 生成合并后的 mask(使用对应颜色的 RGB 值)
52
  merged_mask = np.zeros((height, width, 4), dtype=np.uint8) # RGBA 图像
 
53
  for i, mask in enumerate(masks):
54
  merged_mask[mask, :3] = palette[i] # 使用对应颜色填充 RGB 通道
55
- merged_mask[mask, 3] = 255 # 设置不透明度为完全不透明
56
 
57
  # 将 mask 转换为 PIL 图像(使用对应颜色的 RGB 值)
58
  mask_images = []
59
  for i, mask in enumerate(masks):
60
  mask_array = np.zeros((height, width, 4), dtype=np.uint8) # RGBA 图像
61
  mask_array[mask, :3] = palette[i] # 使用对应颜色填充 RGB 通道
62
- mask_array[mask, 3] = 255 # 设置不透明度为完全不透明
63
  mask_image = Image.fromarray(mask_array, mode='RGBA')
64
  mask_images.append(mask_image)
65
 
 
50
 
51
  # 生成合并后的 mask(使用对应颜色的 RGB 值)
52
  merged_mask = np.zeros((height, width, 4), dtype=np.uint8) # RGBA 图像
53
+ merged_mask[..., 3] = 255 # 设置背景为完全不透明(黑色背景)
54
  for i, mask in enumerate(masks):
55
  merged_mask[mask, :3] = palette[i] # 使用对应颜色填充 RGB 通道
56
+ merged_mask[mask, 3] = 255 # 设置前景为完全不透明
57
 
58
  # 将 mask 转换为 PIL 图像(使用对应颜色的 RGB 值)
59
  mask_images = []
60
  for i, mask in enumerate(masks):
61
  mask_array = np.zeros((height, width, 4), dtype=np.uint8) # RGBA 图像
62
  mask_array[mask, :3] = palette[i] # 使用对应颜色填充 RGB 通道
63
+ mask_array[mask, 3] = 255 # 设置前景为完全不透明
64
  mask_image = Image.fromarray(mask_array, mode='RGBA')
65
  mask_images.append(mask_image)
66