Fix: 변경 마킹 인식 범위 조절 조금 약하게

AI-SIP · Nov 20, 2024 · af472db · af472db
1 parent 722301f
commit af472db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/AIProcessor.py b/app/AIProcessor.py
@@ -107,8 +107,8 @@ def segment_from_boxes(self, image, bbox, save_path=None):
             mask_np = mask.cpu().numpy().astype(np.uint8) * 255  # True는 255, False는 0으로 변환
             mask_np = mask_np.squeeze()
             if i in self.indices:
-                # kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (13, 13))  # Adjust size as needed
-                kernel = np.ones((25, 25), np.uint8)
+                kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (17, 17))  # Adjust size as needed
+                # kernel = np.ones((17, 17), np.uint8)
                 mask_np = cv2.morphologyEx(mask_np, cv2.MORPH_GRADIENT, kernel)
             masks_np = cv2.bitwise_or(masks_np, mask_np)
             # cv2.imwrite(f'mask_box{i}.jpg', masks_np)