dk - edit
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user