diff --git a/main.py b/main.py index bcf8556..e13df68 100644 --- a/main.py +++ b/main.py @@ -10,10 +10,16 @@ from clearml import Task # 1. ClearML 임포트 # 방법 A) 가장 최신 트렌드인 NMS-Free 임베디드 특화 모델 로드 model = YOLO("./weights/yolo26m.pt") # Medium 크기 가중치 자동 다운로드 -task = Task.init( - project_name="Normal_Object_Detection", - task_name="model-yolo26-human" -) + +def init_acai_task(output): + task = Task.init( + project_name="Normal_Object_Detection", + 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: x1 = math.floor(box["x1"]) @@ -51,8 +57,7 @@ def main(image_path: str): } print(json.dumps(output, indent=2, ensure_ascii=False)) - result_data = {"output": output, "status": "PASS"} - task.upload_artifact(name="final_result", artifact_object=result_data) + init_acai_task(output) if __name__ == "__main__": parser = argparse.ArgumentParser()