wangrong
2025-03-19 291afb9c81aae1d15d9e3edcea618d0a2e1fc5a6
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