栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

使用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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/329948.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号