ACAI/model-yolo26-human
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

dk - edit

Browse Source
This commit is contained in:
CHO Daekeun
2026-06-30 10:59:02 +09:00
parent 09242bcd2b
commit f6ef18d66d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+3 -3
View File
@@ -9,6 +9,9 @@ from clearml import Task # 1. ClearML 임포트
# 방법 A) 가장 최신 트렌드인 NMS-Free 임베디드 특화 모델 로드 # 방법 A) 가장 최신 트렌드인 NMS-Free 임베디드 특화 모델 로드
model = YOLO("./weights/yolo26m.pt") # Medium 크기 가중치 자동 다운로드 model = YOLO("./weights/yolo26m.pt") # Medium 크기 가중치 자동 다운로드
# 방법 B) 복잡한 구도나 정밀 식별에 강한 어텐션 기반 모델 로드
# model = YOLO("./weights/yolo12m.pt")
def xyxy_to_xywh(box: dict) -> dict: 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} return {"x": x1, "y": y1, "w": x2 - x1, "h": y2 - y1}
# 방법 B) 복잡한 구도나 정밀 식별에 강한 어텐션 기반 모델 로드
# model = YOLO("yolo12m.pt")
def main(image_path: str): def main(image_path: str):
# uv run main.py --image_url "https://acai.ketidev.kr:20443/detect/image/202606/20260619_145116_image.jpg" # uv run main.py --image_url "https://acai.ketidev.kr:20443/detect/image/202606/20260619_145116_image.jpg"
# 이미지 원격 추론 테스트 (COCO 80종 기본 탐지 가능) # 이미지 원격 추론 테스트 (COCO 80종 기본 탐지 가능)