From a14e6051b183874fe62f84325f190027c43eedec Mon Sep 17 00:00:00 2001 From: zchengrong <124802516+zchengrong@users.noreply.github.com> Date: Wed, 23 Apr 2025 14:40:11 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?= =?UTF-8?q?=EF=BC=89:=20fix=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20=20c?= =?UTF-8?q?lothing=20seg=20=E5=A2=9E=E5=8A=A0=E7=B1=BB=E5=9E=8B=E4=B8=BAsk?= =?UTF-8?q?etch=E7=9A=84=E9=A2=84=E5=A4=84=E7=90=86=20docs=EF=BC=88?= =?UTF-8?q?=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4=EF=BC=89:=20refactor?= =?UTF-8?q?=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20test(=E5=A2=9E=E5=8A=A0?= =?UTF-8?q?=E6=B5=8B=E8=AF=95):?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/clothing_seg/service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/service/clothing_seg/service.py b/app/service/clothing_seg/service.py index 7894bff..46cc444 100644 --- a/app/service/clothing_seg/service.py +++ b/app/service/clothing_seg/service.py @@ -62,7 +62,7 @@ class ClothingSeg: image = data["image"] clothing_result = [] if image_type == "sketch": - seg_mask = get_seg_result(1, image) + seg_mask = get_seg_result(1, image[:, :, :3]) temp = seg_mask != 0.0 mask = (255 * (temp + 0).astype(np.uint8)) x_min, y_min, x_max, y_max = get_bounding_box(mask)