dk - start.py change to main.py

This commit is contained in:
2026-06-29 16:04:33 +09:00
parent d86ef2dbf4
commit c0c3b80c00
+4 -1
View File
@@ -12,7 +12,7 @@ model = YOLO("./weights/yolo26m.pt") # Medium 크기 가중치 자동 다운로
task = Task.init( task = Task.init(
project_name="Normal_Object_Detection", project_name="Normal_Object_Detection",
task_name="YOLOv26m_coc80_Inference_Agent" task_name="YOLOv26m_coco80_Inference_Agent"
) )
def xyxy_to_xywh(box: dict) -> dict: def xyxy_to_xywh(box: dict) -> dict:
@@ -50,6 +50,9 @@ 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"}
task.upload_artifact(name="final_result", artifact_object=result_data)
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--image_url", type=str) parser.add_argument("--image_url", type=str)