上传者: 42131316
|
上传时间: 2021-07-04 15:50:50
|
文件大小: 131KB
|
文件类型: ZIP
0.说明
此存储库包含如何在DeepStream 5.0中运行yolov5模型
1.生成yolov5引擎模型
我们可以使用 yolov5生成引擎模型
重要的提示:
您应该替换tensorrtx / yolov5中的yololayer.cu和hardswish.cu文件
如何运行,以yolov5s为例
- 一种)。 用yolov5s.pt从pytorch生成yolov5s.wts
git clone https://github.com/wang-xinyu/tensorrtx.git
git clone https://github.com/ultralytics/yolov5.git
// download its weights 'yolov5s.pt'
// copy tensorrtx/yolov5/gen_wts.py into ultralytics/yolov5
//