Skip to content

Commit af472db

Browse files
committed
Fix: 변경 마킹 인식 범위 조절 조금 약하게
1 parent 722301f commit af472db

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

app/AIProcessor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,8 +107,8 @@ def segment_from_boxes(self, image, bbox, save_path=None):
107107
mask_np = mask.cpu().numpy().astype(np.uint8) * 255 # True는 255, False는 0으로 변환
108108
mask_np = mask_np.squeeze()
109109
if i in self.indices:
110-
# kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (13, 13)) # Adjust size as needed
111-
kernel = np.ones((25, 25), np.uint8)
110+
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (17, 17)) # Adjust size as needed
111+
# kernel = np.ones((17, 17), np.uint8)
112112
mask_np = cv2.morphologyEx(mask_np, cv2.MORPH_GRADIENT, kernel)
113113
masks_np = cv2.bitwise_or(masks_np, mask_np)
114114
# cv2.imwrite(f'mask_box{i}.jpg', masks_np)

0 commit comments

Comments
 (0)