dk - edit

This commit is contained in:
2026-07-02 11:37:59 +09:00
parent bb4abfa7bc
commit 91539defe2
+6 -7
View File
@@ -6,23 +6,22 @@ from clearml import Task
from start import run from start import run
def upload_acai_task(task: Task, output: dict) -> None:
def init_acai_task(task: Task, output: dict) -> None:
"""UI output 정의에 맞춰 ClearML 아티팩트를 업로드한다.""" """UI output 정의에 맞춰 ClearML 아티팩트를 업로드한다."""
result_data = {"output": output, "status": "PASS"} result_data = {"output": output, "status": "PASS"}
task.upload_artifact(name="final_result", artifact_object=result_data) task.upload_artifact(name="final_result", artifact_object=result_data)
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--image_url", type=str, required=True)
args = parser.parse_args()
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",
) )
parser = argparse.ArgumentParser()
parser.add_argument("--image_url", type=str, required=True)
args = parser.parse_args()
task.connect(parser) task.connect(parser)
output = run(args.image_url) output = run(args.image_url)
init_acai_task(task, output) upload_acai_task(task, output)