网上几乎都是使用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