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

效果也是一样的。

相关推荐
量化风云6 小时前
2026量化新基建(二) - sqlite 与 sqlite-utils
数据库·python·sqlite·量化交易·量化交易课程
程序员老赵6 小时前
ComfyUI Docker 镜像部署指南
人工智能·docker·aigc
用户5191495848456 小时前
Next.js CVE-2025-29927漏洞自动化扫描器
人工智能·aigc
mys55186 小时前
杨建允:AI搜索优化对本地生活、本地服务行业的影响
人工智能·ai搜索优化·ai引擎优化
yuhaiqun19896 小时前
发现前端性能瓶颈的巧妙方法:建立“现象归因→分析定位→优化验证”的闭环思维
前端·经验分享·笔记·python·学习·课程设计·学习方法
Python大数据分析@6 小时前
使用Dify搭建工作流,实现自动化商品采集分析
运维·python·自动化·网络爬虫
code tsunami6 小时前
如何将 Helium 与 CapSolver 集成,实现无缝 CAPTCHA 自动化解决
运维·数据库·人工智能·爬虫·python·自动化
AI科技星6 小时前
空间螺旋电磁耦合常数 Z‘:拨开迷雾,让电磁力变得直观易懂
服务器·人工智能·科技·算法·生活
玄同7656 小时前
Python 异常捕获与处理:从基础语法到工程化实践的万字深度指南
开发语言·人工智能·python·自然语言处理·正则表达式·nlp·知识图谱
云说智树6 小时前
从硅基劳动力到智能工厂:研华iFactory.AI Agent重塑制造新质生产力
人工智能