IQ.Pilot Release Commit @ 8627e35

tinygrad_repo/examples/yolov8-onnx.py

@@ -0,0 +1,13 @@
+#!/usr/bin/env python3
+import os
+from ultralytics import YOLO
+from pathlib import Path
+from tinygrad.nn.onnx import OnnxRunner
+from extra.onnx_helpers import get_example_inputs
+
+os.chdir("/tmp")
+if not Path("yolov8n-seg.onnx").is_file():
+  model = YOLO("yolov8n-seg.pt")
+  model.export(format="onnx", imgsz=[480,640])
+run_onnx = OnnxRunner("yolov8n-seg.onnx")
+run_onnx(get_example_inputs(run_onnx.graph_inputs), debug=True)