19 lines
563 B
Python
19 lines
563 B
Python
from ultralytics import YOLO
|
||
|
||
'''
|
||
yolo官方导出的onnx模型转换为rknn模型后并不能使用.
|
||
'''
|
||
# 加载模型
|
||
model = YOLO("/home/admin-root/haotian/rk3588/pytorch模型转rknn/models/yolov8m.pt")
|
||
|
||
# 导出 ONNX（关键参数！）
|
||
model.export(
|
||
format="onnx",
|
||
imgsz=(640, 640), # 固定输入尺寸
|
||
opset=12, # ONNX 算子版本 >=12
|
||
simplify=True, # 简化模型
|
||
dynamic=False, # 禁用动态轴
|
||
nms=False, # 移除输出端的 NMS 层（RKNN 不支持）
|
||
)
|
||
|
||
# yolo导出的模型 |