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

dk - edit

Browse Source
This commit is contained in:
Daekeun
2026-06-29 17:43:48 +09:00
parent 3bdff0b205
commit 3b18e6053b
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+7 -2
View File
@@ -10,11 +10,17 @@ from clearml import Task # 1. ClearML 임포트
# 방법 A) 가장 최신 트렌드인 NMS-Free 임베디드 특화 모델 로드 # 방법 A) 가장 최신 트렌드인 NMS-Free 임베디드 특화 모델 로드
model = YOLO("./weights/yolo26m.pt") # Medium 크기 가중치 자동 다운로드 model = YOLO("./weights/yolo26m.pt") # Medium 크기 가중치 자동 다운로드
def init_acai_task(output):
task = Task.init( task = Task.init(
project_name="Normal_Object_Detection", project_name="Normal_Object_Detection",
task_name="model-yolo26-human" task_name="model-yolo26-human"
) )
result_data = {"output": output, "status": "PASS"}
task.upload_artifact(name="final_result", artifact_object=result_data)
def xyxy_to_xywh(box: dict) -> dict: def xyxy_to_xywh(box: dict) -> dict:
x1 = math.floor(box["x1"]) x1 = math.floor(box["x1"])
y1 = math.floor(box["y1"]) y1 = math.floor(box["y1"])
@@ -51,8 +57,7 @@ def main(image_path: str):
} }
print(json.dumps(output, indent=2, ensure_ascii=False)) print(json.dumps(output, indent=2, ensure_ascii=False))
result_data = {"output": output, "status": "PASS"} init_acai_task(output)
task.upload_artifact(name="final_result", artifact_object=result_data)
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()