dk - start.py change to main.py
This commit is contained in:
+4
-1
@@ -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)
|
||||
Reference in New Issue
Block a user