Update SegCloth.py

SegCloth.py

@@ -52,21 +52,21 @@ def segment_clothing(img, clothes=["Hat", "Upper-clothes", "Skirt", "Pants", "Dr
             # Application du masque sur l'image vide
             empty_image.paste(segmented_part, mask=mask_image)
             
-            # Déterminer la bounding box du masque
-            bbox = mask_image.getbbox()
-            if bbox:
-                # Ajouter la marge autour de la bounding box
-                left, top, right, bottom = bbox
-                left = max(0, left - margin)
-                top = max(0, top - margin)
-                right = min(img.width, right + margin)
-                bottom = min(img.height, bottom + margin)
-                
-                # Recadrer l'image à la taille du masque avec la marge
-                cropped_image = empty_image.crop((left, top, right, bottom))
-                
-                # Vérifier si l'image recadrée est significative
-                if is_image_significant(cropped_image):
+            # Vérifier si l'image est significative avant le recadrage
+            if is_image_significant(empty_image):
+                # Déterminer la bounding box du masque
+                bbox = mask_image.getbbox()
+                if bbox:
+                    # Ajouter la marge autour de la bounding box
+                    left, top, right, bottom = bbox
+                    left = max(0, left - margin)
+                    top = max(0, top - margin)
+                    right = min(img.width, right + margin)
+                    bottom = min(img.height, bottom + margin)
+                    
+                    # Recadrer l'image à la taille du masque avec la marge
+                    cropped_image = empty_image.crop((left, top, right, bottom))
+                    
                     # Encodage de l'image recadrée en base64
                     imageBase64 = encode_image_to_base64(cropped_image)
                     result_images.append(imageBase64)