You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
524 B
17 lines
524 B
6 days ago
|
#!/usr/bin/env python3
|
||
|
|
import os
|
||
|
|
from ultralytics import YOLO
|
||
|
|
import onnx
|
||
|
|
from pathlib import Path
|
||
|
|
from tinygrad.frontend.onnx import OnnxRunner
|
||
|
|
from extra.onnx_helpers import get_example_inputs
|
||
|
|
from tinygrad.tensor import Tensor
|
||
|
|
|
||
|
|
os.chdir("/tmp")
|
||
|
|
if not Path("yolov8n-seg.onnx").is_file():
|
||
|
|
model = YOLO("yolov8n-seg.pt")
|
||
|
|
model.export(format="onnx", imgsz=[480,640])
|
||
|
|
onnx_model = onnx.load(open("yolov8n-seg.onnx", "rb"))
|
||
|
|
run_onnx = OnnxRunner(onnx_model)
|
||
|
|
run_onnx(get_example_inputs(run_onnx.graph_inputs), debug=True)
|