ultralytics中rtdetr的.pt模子转onnx

1. 将训练的.pt模子转换为onnx模子

1. import os
2. print(os.getcwd())
4. from ultralytics import RTDETR
6. # Load a COCO-pretrained RT-DETR-l model
7. model_pth = r"E:\02.Projects\01.ultralytics\ultralytics\models\rtdetr\adas\single_category_vehicle-rtdetr\weights\best.pt"
8. model = RTDETR(model_pth)   # "rtdetr-l.pt"
9. # Display model information (optional)
10. model.info()
11. #model.export(format='onnx',opset=17, simplify=True, dynamic=False, imgsz=(640, 640))  # default, opset=19
12. model.export(format='onnx')  # default, opset=19
14. print('Done.')

复制代码

2. 验证转换后的模子是否正确

1. import onnx
2. import onnxruntime as ort
4. # 加载 ONNX 模型
5. onnx_model = onnx.load('rtdetr-l.onnx')
7. # 检查模型是否正确
8. onnx.checker.check_model(onnx_model)
10. # 使用 onnxruntime 运行模型
11. ort_session = ort.InferenceSession('rtdetr-l.onnx')
13. # 准备输入数据（根据模型的输入要求进行准备）
14. # outputs = ort_session.run(None, {input_name: input_data})
16. print('Done.')

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。