公告板
版本库
filestore
活动
搜索
登录
main
/
VideoPipe
base for mixc
概况
操作记录
提交次数
目录
文档
派生
对比
blame
|
历史
|
原始文档
add samples
wangrong
2025-03-19
291afb9c81aae1d15d9e3edcea618d0a2e1fc5a6
[VideoPipe.git]
/
third_party
/
trt_yolov8
/
samples
/
gen_onnx.py
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