faster when alpha mask = 1

Author: Koldim2001

patched_yolo_infer/functions_extra.py

@@ -355,9 +355,12 @@ def visualize_results(
             )
 
             if fill_mask:
-                color_mask = np.zeros_like(img)
-                color_mask[mask_resized > 0] = color
-                labeled_image = cv2.addWeighted(labeled_image, 1, color_mask, alpha, 0)
+                if alpha == 1:
+                    cv2.fillPoly(labeled_image, pts=mask_contours, color=color)
+                else:
+                    color_mask = np.zeros_like(img)
+                    color_mask[mask_resized > 0] = color
+                    labeled_image = cv2.addWeighted(labeled_image, 1, color_mask, alpha, 0)
 
             cv2.drawContours(labeled_image, mask_contours, -1, color, thickness)
 
@@ -366,7 +369,7 @@ def visualize_results(
                 points = np.array(polygons[i].reshape((-1, 1, 2)), dtype=np.int32)
                 cv2.drawContours(labeled_image, [points], -1, color, thickness)
                 if fill_mask:
-                    cv2.fillPoly(labeled_image, pts=[points], color=(*color, alpha))
+                    cv2.fillPoly(labeled_image, pts=[points], color=color)
 
         # Write class label
         if show_boxes: