muk42 commited on
Commit
1a5b77a
·
1 Parent(s): 1a5251c
Files changed (1) hide show
  1. inference_tab/inference_logic.py +2 -2
inference_tab/inference_logic.py CHANGED
@@ -291,8 +291,8 @@ def extractSegments(image_path, min_size=500, margin=100):
291
  # Apply mask to original image
292
  shaded = cropped_image.copy()
293
  overlay = cropped_image.copy()
294
- overlay[cropped_mask == 1] = (0, 0, 255)
295
- alpha = 0.5
296
  shaded = cv2.addWeighted(overlay, alpha, shaded, 1 - alpha, 0)
297
 
298
  # Save the masked image
 
291
  # Apply mask to original image
292
  shaded = cropped_image.copy()
293
  overlay = cropped_image.copy()
294
+ overlay[cropped_mask == 1] = (255, 200, 100)
295
+ alpha = 0.75
296
  shaded = cv2.addWeighted(overlay, alpha, shaded, 1 - alpha, 0)
297
 
298
  # Save the masked image