yolov5 通过视频进行目标检测

打开yolov5-master 文件夹,可以看到一个名为data 的文件夹,在data 中创建一个新的文件夹,命名为videos

打开yolov5-master中的detect.py可以看到一行代码(大概在245行左右)为

python 复制代码
parser.add_argument('--source', type=str, default=ROOT / 'data/images', help='file/dir/URL/glob/screen/0(webcam)')

我们先把它注释掉(在前面加个#)

然后复制它,在它原本的下面添加这样一行代码

python 复制代码
parser.add_argument('--source', type=str, default=ROOT / 'data/voides', help='file/dir/URL/glob/screen/0(webcam)')

可以看到我们只把images改成了voides,意思是现在让它去识别videos下的所有东西,如果你只想要识别videos中的一个视频(设这个视频为xxx.mp4)可以这样写,

python 复制代码
parser.add_argument('--source', type=str, default=ROOT / 'data/videos/xxx.mp4', help='file/dir/URL/glob/screen/0(webcam)')

我们把需要检测的视频放在voides中,然后Run一下就可以检测视频了

这里我用一下这个视频行人检测测试视频_哔哩哔哩_bilibili

行人检测测试视频

可以看到前面(1395/7581),其实就是它自己把视频逐帧的当做图片去识别。

跑完后,我们可以看到,跑完后的视频的地址。

打开视频live.csdn.net/v/342351

如果跑的时间太长,我们想要让它一边跑,一边看呢?

为了实现这个步骤,我们要先开启显示预览的功能,

在大概253行可以看到这行代码,这个显示预览的功能默认是关闭的。

python 复制代码
    parser.add_argument('--view-img', action='store_true', help='show results')

默认是这样的

python 复制代码
 parser.add_argument('--view-img', action='store_true', help='show results', default='False')

打开其实就很简单了,把false改成true

python 复制代码
 parser.add_argument('--view-img', action='store_true', help='show results', default='True')

改完再去Run就会弹出这个预览的界面了。

除此之外还有一个不用修改代码就能显示预览的方法,

在左上角的Run点进去 有个Edit Configurations点进去

输入--view-img

效果也是一样的。

相关推荐
星越华夏1 天前
python中四种获取文件后缀名的方法
开发语言·python
lunzi_08261 天前
【学习笔记】《Python编程 从入门到实践》第9章:类、继承、组合与面向对象编程
笔记·python·学习
思陌Ai算法定制1 天前
【心血管影像AI预测预后】心肌梗死后心脏MRI如何更早识别高风险患者?
人工智能·影像组学·心血管影像·心脏mri·心肌梗死·stemi·微循环阻塞
大蚂蚁2号1 天前
本地批量音视频转文本免费工具
python·音视频·开源软件
云烟成雨TD1 天前
Agent Scope Java 2.x 系列【6】消息层
java·人工智能·agent
云烟成雨TD1 天前
Spring AI Alibaba 1.x 系列【74】Agentic RAG 与混合 RAG
java·人工智能·spring
Tiansan66661 天前
郑州AI问答推广:2家优质服务商剖析
人工智能·郑州ai问答推广2家
“码”力全开1 天前
解耦异构算力:基于 Docker 与边缘计算的 GB28181/RTSP 企业级 AI 视频管理平台架构设计(含源码交付)
人工智能·docker·边缘计算
云烟成雨TD1 天前
Spring AI Alibaba 1.x 系列【79】图执行生命周期的可观测性基础设施
java·人工智能·spring
kishu_iOS&AI1 天前
LLM —— Milvmus向量数据库
数据库·人工智能·milvus