yolov5 版本
ethannie11 opened this issue · 2 comments
ethannie11 commented
您好 当将训练好的模型用tensorrt加速时出现了网络不匹配的问题,请问您当时是用的哪个版本的yolov5呢 ? 是v2.0还是3.0,或者是其他版本的?
DataXujing commented
由于历史原因,该项目构建时还没有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 commented
我们提供了其他方式的TensorRT加速的方式,如有需要我们会将其C++和Python代码提供给您共参考!