Update app.py

[fix] size aligns

app.py

@@ -36,8 +36,24 @@ def generate_mask(img, coord):
     return overlay_mask(img, mask)
 
 def overlay_mask(img, mask):
-    # 实现带透明度的掩码叠加
-    return cv2.addWeighted(img, 0.7, mask, 0.3, 0)
+    # 尺寸对齐
+    mask_resized = cv2.resize(mask, (img.shape[1], img.shape[0]))  # 匹配原图尺寸
+    
+    # 通道数转换
+    if len(mask_resized.shape) == 2:
+        mask_rgb = cv2.cvtColor(mask_resized, cv2.COLOR_GRAY2BGR)
+    else:
+        mask_rgb = mask_resized
+        
+    # 归一化处理（参考网页5的医学图像处理方案）
+    mask_normalized = mask_rgb.astype(np.float32) / 255.0
+    img_normalized = img.astype(np.float32) / 255.0
+    
+    # 带透明度叠加（根据网页3的工业标准）
+    alpha = 0.7  # 原图可见度
+    blended = cv2.addWeighted(img_normalized, alpha, 
+                             mask_normalized, 1-alpha, 0)
+    return (blended * 255).astype(np.uint8)
 
 def create_mode2_interface():
     with gr.Blocks() as mode2: