使用DeepStream5.0-python-API 部署yolov5

网上几乎都是使用c来部署yolov5的,使用python-API的我还没找到,github上也没找到用python编写的代码来启动yolov5,好在经过对c部署yolov5的研究,实现了使用DeepStream5.0-python-API 部署yolov5。

我们已经在之前部署过python-API-yolov3,目录sources/deepstream_python_apps-1.0/apps/deepstream-yolov3-python。也部署过C-API-yolov5,一拍即合。

在sources/deepstream_python_apps-1.0/apps下新建deepstream-yolov5-python文件夹。

将下载的DeepStream-Yolo/external/yolov5-5.0 文件夹里面的文件复制到创建的 deepstream-yolov5-python 文件夹下,将生成的yolov5s.engine文件也放到该文件夹下。

编译:
在deepstream-yolov5-python目录下运行

CUDA_VER=10.2 make -C nvdsinfer_custom_impl_Yolo

将sources/deepstream_python_apps-1.0/apps/deepstream-yolov3-python/deepstream_test_3.py复制到deepstream-yolov5-python目录下,重命名为deepstream_yolov5.py。

修改deepstream_yolov5.py的内容:
第292行的config_infer_primary_yolov3.txt改为config_infer_primary.txt。

运行:

python3 deepstream_yolov5.py rtsp://admin:tensorsight666@192.168.8.196:554/h265/ch1/main/av_stream

文件打包下载:
https://download.csdn.net/download/W1995S/33163540


版权声明:本文为W1995S原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。