diff --git a/modules/masking.py b/modules/masking.py index ca8db1040..76bd93137 100644 --- a/modules/masking.py +++ b/modules/masking.py @@ -340,7 +340,7 @@ def get_mask(input_image: gr.Image, input_mask: gr.Image): largest_size = cv2.contourArea(contours[0]) if len(contours) > 0 else 0 for i, contour in enumerate(contours): area_size = cv2.contourArea(contour) - luminance = int(255.0 * area_size / largest_size) + luminance = int(255.0 * area_size / largest_size) if largest_size > 0 else 0 if luminance < 1: break cv2.drawContours(output_mask, contours, i, (luminance), -1)