yolov5 版本

[ethannie11]

trafficstars

您好 当将训练好的模型用tensorrt加速时出现了网络不匹配的问题，请问您当时是用的哪个版本的yolov5呢 ？ 是v2.0还是3.0，或者是其他版本的？

[DataXujing]

由于历史原因，该项目构建时还没有yolov5 2.0,3.0,4.0,5.0, 因此该项目是基于yolo v5 1.0的，但是训练方式是一致的，关于TensorRT加速部分，建议使用TensorRT 7.1以后的版本，本项目提供的TensorRT加速方式来源于https://github.com/wang-xinyu/tensorrtx,请参考该项目按照自己版本进行加速

[DataXujing]

我们提供了其他方式的TensorRT加速的方式，如有需要我们会将其C++和Python代码提供给您共参考！