公告板
版本库
filestore
活动
搜索
登录
main
/
VideoPipe
base for mixc
概况
操作记录
提交次数
目录
文档
派生
对比
blame
|
历史
|
原始文档
add person gathering
wangrong
2025-01-18
8b0e31121a9a29b92a41fc575138a2e9b4c12d8e
[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