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 分钟前
GLM4.6多工具协同开发实践:AI构建智能任务管理系统的完整指南
人工智能·python·状态模式·ai编程·glm·分类系统·开发架构
川西胖墩墩3 分钟前
中文PC端跨职能流程图模板免费下载
大数据·论文阅读·人工智能·架构·流程图
Keep_Trying_Go14 分钟前
MaskGIT掩码生成图算法详解(MaskGIT: Masked Generative Image Transformer)
人工智能·深度学习·transformer
致Great16 分钟前
大模型对齐核心技术:从第一性原理完整推导 PPO 算法!
人工智能·算法·大模型·agent·智能体
Darken0320 分钟前
基于STM32---编码器测速(利用GPIO模拟脉冲信号)
人工智能·stm32·串口助手·gpio模拟编码器
Mintopia20 分钟前
🪄 生成式应用的 **前端 orchestration 层(编排层)指南**
人工智能·llm·aigc
雍凉明月夜25 分钟前
深度学习之常用归一化(Normalization)
人工智能·深度学习·计算机视觉
沃达德软件25 分钟前
视频标注技术全解析
人工智能·目标检测·计算机视觉·视觉检测·音视频·实时音视频·视频编解码
Buxxxxxx28 分钟前
DAY 44 简单CNN
人工智能·神经网络·cnn
GEO AI搜索优化助手30 分钟前
从传统SEO到生成式AI搜索优化的战略转型
人工智能·搜索引擎·生成式引擎优化·ai优化·geo搜索优化