Ayesha-Majeed commited on
Commit
f769ef6
·
verified ·
1 Parent(s): 5096687

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -62,17 +62,16 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
62
  CONF = 0.45
63
 
64
  def apply_mask_overlay(img_rgb, mask_bool, color=(0, 215, 255), alpha=0.4):
65
- # 1. Cinematic Background (Bokeh/Blur effect + darkened)
66
- blurred_bg = cv2.GaussianBlur(img_rgb, (35, 35), 0)
67
- background = cv2.addWeighted(blurred_bg, 0.6, np.zeros_like(img_rgb), 0.4, 0)
68
 
69
- # 2. Keep the segmented mask area perfectly sharp and tint it
70
  tinted_sharp = img_rgb.copy()
71
  tinted_sharp[mask_bool] = color
72
  tinted_sharp = cv2.addWeighted(tinted_sharp, alpha, img_rgb, 1 - alpha, 0)
73
 
74
- # 3. Combine: Blurred background + Sharp highlighted object
75
- blended = np.where(mask_bool[:, :, None], tinted_sharp, background)
76
 
77
  # 4. Multi-layered Glowing Edge for the mask
78
  mask_img = (mask_bool * 255).astype(np.uint8)
 
62
  CONF = 0.45
63
 
64
  def apply_mask_overlay(img_rgb, mask_bool, color=(0, 215, 255), alpha=0.4):
65
+ # 1. Darken the background (50% brightness, no blur)
66
+ dark_bg = cv2.addWeighted(img_rgb, 0.5, np.zeros_like(img_rgb), 0.5, 0)
 
67
 
68
+ # 2. For the mask area, keep original brightness and tint it
69
  tinted_sharp = img_rgb.copy()
70
  tinted_sharp[mask_bool] = color
71
  tinted_sharp = cv2.addWeighted(tinted_sharp, alpha, img_rgb, 1 - alpha, 0)
72
 
73
+ # 3. Combine: Dark background outside, Bright tinted object inside
74
+ blended = np.where(mask_bool[:, :, None], tinted_sharp, dark_bg)
75
 
76
  # 4. Multi-layered Glowing Edge for the mask
77
  mask_img = (mask_bool * 255).astype(np.uint8)