GitHub - DanaHan/Yolov5-in-Deepstream-5.0: Describe how to use yolov5 in Deepstream 5.0
#TODO: 视频接入与deepstreamer
使用:
This Repos contains how to run yolov5 model in DeepStream 5.0
1.Geneate yolov5 engine model
We can use https://github.com/wang-xinyu/tensorrtx yolov5 to generate engine model
Important Note:
You should replace yololayer.cu and hardswish.cu file in tensorrtx/yolov5
How to Run, yolov5s as example
-- a). generate yolov5s.wts from pytorch with yolov5s.pt
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
// ensure the file name is yolov5s.pt and yolov5s.wts in gen_wts.py
// go to ultralytics/yolov5
python g