diff --git a/main.py b/main.py index 5370755..302a20f 100644 --- a/main.py +++ b/main.py @@ -9,6 +9,9 @@ from clearml import Task # 1. ClearML 임포트 # 방법 A) 가장 최신 트렌드인 NMS-Free 임베디드 특화 모델 로드 model = YOLO("./weights/yolo26m.pt") # Medium 크기 가중치 자동 다운로드 +# 방법 B) 복잡한 구도나 정밀 식별에 강한 어텐션 기반 모델 로드 +# model = YOLO("./weights/yolo12m.pt") + def xyxy_to_xywh(box: dict) -> dict: @@ -19,9 +22,6 @@ def xyxy_to_xywh(box: dict) -> dict: return {"x": x1, "y": y1, "w": x2 - x1, "h": y2 - y1} -# 방법 B) 복잡한 구도나 정밀 식별에 강한 어텐션 기반 모델 로드 -# model = YOLO("yolo12m.pt") - def main(image_path: str): # uv run main.py --image_url "https://acai.ketidev.kr:20443/detect/image/202606/20260619_145116_image.jpg" # 이미지 원격 추론 테스트 (COCO 80종 기본 탐지 가능)