From 09242bcd2b7c596124442c350d4c8bbb19a6342c Mon Sep 17 00:00:00 2001 From: Daekeun Date: Mon, 29 Jun 2026 18:25:58 +0900 Subject: [PATCH] dk - edit --- README.md | 9 +++++---- main.py | 1 + 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index b7e80b0..fb8e1fa 100644 --- a/README.md +++ b/README.md @@ -20,8 +20,8 @@ uv run main.py --image_url "https://example.com/image.jpg" ``` - `--image_url`: 추론할 이미지 URL (필수) -- 모델: `weights/yolo26m.pt` (Ultralytics YOLO) -- 탐지 결과가 그려진 이미지는 `*_marked.jpg`로 저장됩니다. +- 모델: `weights/yolo26m.pt` (Ultralytics YOLO, COCO 80종 탐지) +- 이미지는 `image.jpg`로 저장되고, 탐지 결과 이미지는 `*_marked.jpg`로 저장됩니다. ## 출력 @@ -34,13 +34,14 @@ uv run main.py --image_url "https://example.com/image.jpg" | `shape` | 이미지 크기 (width, height) | | `speed_ms` | 추론 속도 (ms) | | `detections` | 탐지 객체 목록 (클래스, 신뢰도, `xywh` 박스) | +| `image_url` | 이미지 경로 | ## ClearML -실행 시 ClearML Task가 생성됩니다. +스크립트 실행 시 ClearML Task가 초기화되고, 추론 완료 후 결과가 아티팩트로 업로드됩니다. - **Project**: `Normal_Object_Detection` - **Task**: `model-yolo26-human` -- **Artifact**: `final_result` (추론 결과 JSON) +- **Artifact**: `final_result` (`output`, `status: PASS`) ClearML 서버 연결 설정이 필요합니다 (`clearml-init`). diff --git a/main.py b/main.py index 35fa166..5370755 100644 --- a/main.py +++ b/main.py @@ -44,6 +44,7 @@ def main(image_path: str): "shape": {"height": result.orig_shape[0], "width": result.orig_shape[1]}, "speed_ms": result.speed, "detections": detections, + "image_url": image_path, } print(json.dumps(output, indent=2, ensure_ascii=False))