update garment pil

app.py

@@ -102,15 +102,31 @@ def apply_parsing_white_mask_to_person_cv2(
     person_pil: Image.Image,
     parsing_img: Image.Image
 ) -> np.ndarray:
+    """
+    person_pil(RGB) 크기에 parsing_img(L) 마스크를 맞춰서
+    흰색(255) 영역만 person을 남기고 나머지는 흰색 배경으로 만드는 함수.
+
+    - parsing_img는 person 크기에 반드시 맞춰야 함 (NEAREST)
+    """
     person_rgb = np.array(person_pil.convert("RGB"), dtype=np.uint8)
+
+    # parsing 마스크 (L)
     mask = np.array(parsing_img.convert("L"), dtype=np.uint8)
-    white_mask = mask == 255
+
+    # ✅ 핵심: 크기 불일치 해결 (H,W) 맞춤
+    if mask.shape[0] != person_rgb.shape[0] or mask.shape[1] != person_rgb.shape[1]:
+        mask = cv2.resize(mask, (person_rgb.shape[1], person_rgb.shape[0]), interpolation=cv2.INTER_NEAREST)
+
+    white_mask = (mask == 255)
+
     result_rgb = np.full_like(person_rgb, 255, dtype=np.uint8)
     result_rgb[white_mask] = person_rgb[white_mask]
+
     result_bgr = cv2.cvtColor(result_rgb, cv2.COLOR_RGB2BGR)
     return result_bgr
 
 
+
 def compute_hw_from_person(person_path: str):
     img = _imread_or_raise(person_path)
     orig_h, orig_w = img.shape[:2]