tensorrt高级2：YoloV5 模型导出、编译到推理(源码讲解）

文章目录

0 前言

学习yolov5如何导出模型并利用起来：

  • 1.修改export_onnx时导出参数，使得动态维度指定为batch,去掉width 和height的指定。
  • 2.导出时，对yolo.py进行修改，使得后处理能够简化，并将anchor合并到onnx中
  • 3.预处理部分采用warpaffine,对图像进行平移和缩放

1.修改onnx

导出并可视化onnx模型

1）下载yolov5-6.0代码：https://github.com/ultralytics/yolov5/tree/v6.0

版权声明：本文为weixin_38346042原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。