Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|