fix visual v1.2.2

Author: Koldim2001

patched_yolo_infer/functions_extra.py

@@ -361,20 +361,19 @@ def visualize_results(
                     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)
 
         elif segment and len(polygons) > 0:
             if len(polygons[i]) > 0:
                 points = np.array(polygons[i].reshape((-1, 1, 2)), dtype=np.int32)
-                cv2.drawContours(labeled_image, [points], -1, color, thickness)
                 if fill_mask:
                     if alpha == 1:
                         cv2.fillPoly(labeled_image, pts=[points], color=color)
                     else:
                         mask_from_poly = np.zeros_like(img)
                         color_mask_from_poly = cv2.fillPoly(mask_from_poly, pts=[points], color=color)
                         labeled_image = cv2.addWeighted(labeled_image, 1, color_mask_from_poly, alpha, 0)
+                cv2.drawContours(labeled_image, [points], -1, color, thickness)
 
         # Write class label
         if show_boxes:

setup.py

@@ -8,7 +8,7 @@
     long_description = "\n" + fh.read()
 
 
-VERSION = '1.2.1'
+VERSION = '1.2.2'
 DESCRIPTION = '''YOLO-Patch-Based-Inference for detection/segmentation of small objects in images.'''
 
 setup(