From 5086a7a8b0ec1a12a226ff6d340c36a5651645b7 Mon Sep 17 00:00:00 2001 From: Phoenix Date: Tue, 1 Jul 2025 21:31:02 +0300 Subject: [PATCH] Update SegCloth.py --- SegCloth.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/SegCloth.py b/SegCloth.py index fa95e30..d07ed18 100644 --- a/SegCloth.py +++ b/SegCloth.py @@ -48,7 +48,8 @@ def segment_clothing(img, category): # Create the final image by pasting the gray overlay onto the original using the clothing mask fine_mask_clothes = img_rgb.copy() fine_mask_clothes.paste(gray_overlay, mask=binary_mask) - + binary_mask.save("binary_mask.jpg") + fine_mask_clothes.save("fine_mask_clothes.jpg") return binary_mask, fine_mask_clothes