IT评测·应用市场-qidao123.com

标题: ultralytics中rtdetr的.pt模子转onnx [打印本页]
作者: 圆咕噜咕噜    时间: 2025-1-2 15:38
标题: 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企服之家，中国第一个企服评测及商务社交产业平台。



欢迎光临 IT评测·应用市场-qidao123.com (https://dis.qidao123.com/) Powered by Discuz! X3.4