djl234 commited on
Commit
e454f1a
·
verified ·
1 Parent(s): 209dd06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -65,13 +65,13 @@ def overlay_mask(img, mask):
65
  mask_rgb = mask_resized
66
 
67
  # 归一化处理
68
- return cv2.addWeighted(
69
  img.astype(np.float32)/255,
70
  0.7,
71
  mask_rgb.astype(np.float32)/255,
72
  0.3,
73
  0
74
- ).astype(np.uint8)
75
 
76
  def create_mode2_interface():
77
  with gr.Blocks() as mode2:
 
65
  mask_rgb = mask_resized
66
 
67
  # 归一化处理
68
+ return (cv2.addWeighted(
69
  img.astype(np.float32)/255,
70
  0.7,
71
  mask_rgb.astype(np.float32)/255,
72
  0.3,
73
  0
74
+ )*255).astype(np.uint8)
75
 
76
  def create_mode2_interface():
77
  with gr.Blocks() as mode2: