Spaces:
Running
Running
Update SegCloth.py
Browse files- SegCloth.py +1 -1
SegCloth.py
CHANGED
@@ -35,7 +35,7 @@ def segment_clothing(img, clothes=["Hat", "Upper-clothes", "Skirt", "Pants", "Dr
|
|
35 |
cv2.drawContours(final_mask, contours, -1, (255), thickness=cv2.FILLED)
|
36 |
|
37 |
# Apply Gaussian blur to smooth edges and reduce noise
|
38 |
-
final_mask = cv2.GaussianBlur(final_mask, (7, 7), 0)
|
39 |
|
40 |
# Convert mask to binary (0 or 255) if needed for alpha channel
|
41 |
_, final_mask = cv2.threshold(final_mask, 127, 255, cv2.THRESH_BINARY)
|
|
|
35 |
cv2.drawContours(final_mask, contours, -1, (255), thickness=cv2.FILLED)
|
36 |
|
37 |
# Apply Gaussian blur to smooth edges and reduce noise
|
38 |
+
#final_mask = cv2.GaussianBlur(final_mask, (7, 7), 0)
|
39 |
|
40 |
# Convert mask to binary (0 or 255) if needed for alpha channel
|
41 |
_, final_mask = cv2.threshold(final_mask, 127, 255, cv2.THRESH_BINARY)
|