yoloV5的.pt模型转换为.engine加速模型,通过Deepstream加速
一,环境(最新版本的环境):
deepstream:6.1
jetpack:5.0.1DP,其中自带CUDA11.4
技术路线:.pt->.wts/.cfg->.engine
两个重要项目地址:
1,https://github.com/ultralytics/yolo5;
2,https://github.com/marcoslucianops/DeepStream-Yolo
二,如何选择deepstream和yolov5的兼容版本
A,版本选择
1,yolov5下载git地址(注意版本的选择,而且它不能运行deepstream生成的engine,代码作者在优化合并,估计以后就不会有这么多问题了):
yolov5-5.0(.pt->.wts and .cfg)
网址:https://github.com/ultralytics/yolo5
you need otice:yolov5-6.0 complete wts and cfg to