diff --git a/start.py b/main.py similarity index 92% rename from start.py rename to main.py index c3e77fa..4b00e7b 100644 --- a/start.py +++ b/main.py @@ -12,7 +12,7 @@ model = YOLO("./weights/yolo26m.pt") # Medium 크기 가중치 자동 다운로 task = Task.init( project_name="Normal_Object_Detection", - task_name="YOLOv26m_coc80_Inference_Agent" + task_name="YOLOv26m_coco80_Inference_Agent" ) 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)) + result_data = {"output": output, "status": "PASS"} + task.upload_artifact(name="final_result", artifact_object=result_data) + if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument("--image_url", type=str)