Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

yolov5 版本 #31

Open
ethannie11 opened this issue May 31, 2021 · 2 comments
Open

yolov5 版本 #31

ethannie11 opened this issue May 31, 2021 · 2 comments

Comments

@ethannie11
Copy link

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

@DataXujing
Copy link
Owner

由于历史原因,该项目构建时还没有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
Copy link
Owner

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

@DataXujing DataXujing pinned this issue Jun 23, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants