From 03b3890a2dc28bad6d6c33b1bb48e747b8d265f2 Mon Sep 17 00:00:00 2001 From: semnisem Date: Wed, 20 Nov 2024 01:50:28 +0900 Subject: [PATCH] =?UTF-8?q?HotFix:=20=EB=B2=84=EA=B7=B8=20=EC=88=98?= =?UTF-8?q?=EC=A0=95=20=EC=9E=85=EB=A0=A5=20=ED=95=84=ED=84=B0=EB=A7=81=20?= =?UTF-8?q?=EC=A0=9C=EA=B1=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/AIProcessor.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/app/AIProcessor.py b/app/AIProcessor.py index 4136160..1462522 100644 --- a/app/AIProcessor.py +++ b/app/AIProcessor.py @@ -58,7 +58,6 @@ def object_detection(self, img): results = self.yolo_model.predict(source=img, imgsz=640, device=self.device, iou=0.3, conf=0.3) bbox = results[0].boxes.xyxy.tolist() - logging.info(f'1차 마스킹 - 바운딩 박스 생성 완료, {len(bbox)}개') return bbox def segment_from_points(self, image, user_points, user_labels, bbox, save_path=None): @@ -70,8 +69,8 @@ def segment_from_points(self, image, user_points, user_labels, bbox, save_path= point_labels=input_label, multimask_output=False, )''' - filtered_points, filtered_labels = self.remove_points_in_bboxes(user_points, user_labels, bbox) - logging.info(f'2차 마스킹 - 사용자 입력 필터링: from {len(user_points)}개 to {len(filtered_points)}개') + # filtered_points, filtered_labels = self.remove_points_in_bboxes(user_points, user_labels, bbox) + # logging.info(f'2차 마스킹 - 사용자 입력 필터링: from {len(user_points)}개 to {len(filtered_points)}개') # results = self.sam_model.predict(source=image, points=filtered_points, labels=filtered_labels) results = self.sam_model.predict(source=image, bboxes=[user_points])