blend gray image

segment_utils.py

@@ -116,7 +116,9 @@ def segment_image_with_color(input_image, color, alpha, category, input_size, ma
     cv2_target_hsv[..., 1] = targetHsv[1]
 
     cv2_target_bgr = cv2.cvtColor(cv2_target_hsv, cv2.COLOR_HSV2BGR)
-    cv2_image_w = cv2.addWeighted(cv2_target_bgr, alpha, cv2_image, 1 - alpha, 0)
+    gray_image = cv2.cvtColor(cv2_image, cv2.COLOR_BGR2GRAY)
+    gray_image_3d = cv2.merge([gray_image] * 3)
+    cv2_image_w = cv2.addWeighted(cv2_target_bgr, alpha, gray_image_3d, 1 - alpha, 0)
     cv2_image[target_mask] = cv2_image_w[target_mask]
 
     input_image = Image.fromarray(cv2.cvtColor(cv2_image, cv2.COLOR_BGR2RGB))