Add ClearML main.py wrapper for ACAI

This commit is contained in:
2026-07-16 15:56:30 +09:00
commit e472b80387
12 changed files with 2016 additions and 0 deletions
+26
View File
@@ -0,0 +1,26 @@
import argparse
from clearml import Task
from start import run
def upload_acai_task(task: Task, output: dict) -> None:
"""UI output 정의에 맞춰 업로드"""
result_data = {"output": output, "status": "PASS"}
task.upload_artifact(name="final_result", artifact_object=result_data)
if __name__ == "__main__":
task = Task.init(
project_name="Normal_Object_Detection",
task_name="model-yolo26-human",
)
parser = argparse.ArgumentParser()
parser.add_argument("--image_url", type=str, required=True)
args = parser.parse_args()
task.connect(parser)
output = run(args.image_url)
upload_acai_task(task, output)