Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4959 from QualiaRain/fix/mask-edge-zero-area-divide

Browse Source 
fix(mask): guard zero-area divide in auto-mask Edge mode
This commit is contained in:
Vladimir Mandic
2026-06-22 20:39:08 +02:00
committed by GitHub
parent 0485e8708f ec3d4182cc
commit 3e1df229ba
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/masking.py
+1 -1
View File
@@ -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)