add sample

wangrong

2025-01-23 02819b5c047bb354b0ef2374e7c6a6ac4bbd5bba

1
2
3
4
5
6
7
8
9
10
11
from ultralytics import YOLO 
 
''' 
NOTE: trt_yolov8 do not need ONNX format at all, this script just used for visualizing yolov8 networks on `https://netron.app`.  
''' 
 
# Load a model 
model = YOLO("../../../../vp_data/models/trt/others/yolov8n-seg.pt")  # load a pretrained model (recommended for training) 
 
# Export the model 
path = model.export(format="onnx")  # export the model to ONNX format which could be visualized on netron.app