- 博客(1)
- 收藏
- 关注
RSS订阅原创 yolov5模型的部署之TensorRT
写在前面，利用TensorRT加速推理速度是以时间换取精度的做法，意味着在推理速度上升的同时将会有精度的下降，不过不用太担心，精度下降微乎其微。此外，要有NVIDIA显卡，经测试，CUDA10.2可以支持20系列显卡及以下，30系列显卡需要CUDA11.x的支持，并且目前有bug。 默认你已经完成了yolov5的训练过程并得到了.pt模型权值文件。 本文目的仅是带着走通流程。 注意要对应yolov5和tensorrtx的版本。 我的运行环境(注意OpenCV要选择适合你的visual st.
2021-09-24 22:14:34
2672
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人