有一个需求,要在视频中进行截图,人工处理的话,就是要求选择一些比较精彩的镜头进行手工截取图片,这就意味着,每个视频都要自己去看一遍,然后选择比较好的镜头截图保存。
但是这种方法有很多弊端:
于是迫切需要一个自动化的工具,能够对视频进行截图,按照一定的规则截取,比如按照时间间隔、时间点、帧数等等进行截图,哪怕是截出来的图片可能镜头不是很好,截完图后再去人工筛选也是比较快的。
Python是比较好上手的一个编程语言了,而且最近一直挺火的,主要是其强大的类库,省去了大量的造轮子的过程,让编程更加专注于业务本身,更加专注于实现你的目的。
而且Python这门语言不难,会点计算机知识、会点编程的应该一学都会了。
利用Python进行视频截图,是一件相对比较容易实现的事情,而且一旦写成,一劳永逸,能节省掉后期大量手工截图花的功夫,有那个时间,喝喝茶、看看剧不是挺好吗?
这篇文章目的也是为了照顾各种小白,让大家更好的理解,所以介绍的会比较通俗一些。
好了,那么就让我们开始吧!
这个就不用多说了吧,大家直接去Python官网下载安装包。
然后安装即可。实在不会的话,百度一下有很多安装教程,照葫芦画瓢一学就会。这不是本文的核心,因此不做过多展开了。
这里笔者比较推荐使用vscode和pycharm。
vscode是微软推出的一款强大的编辑器,支持各种语言的开发,以及文本文档的编写。其强大的插件机制,让一款编辑器可以跃身成为一款强大的“IDE”。
pycharm是一款jetbrains公司推出的,专门用于python开发的IDE,非常强大好用。
二者选其一,轻量级的选vscode会比较好。(推荐)
创建一个py脚本文件,取名的话,你可以随意,我这里取名为:video_capture.py
这个脚本文件就是我们要写代码的地方了。
opencv这个类库时需要先安装才能使用。
按 Win + R组合键,调出搜索框,输入cmd,回车打开命令行。
在命令行中输入下面这行内容进行安装:
pip install opencv-python
但是可能会安装失败:
这是因为源的问题,改成阿里源就能安装成功了!
pip install opencv-python -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
嗯,这样就安装成功了。
如果你看不懂,或者实在不想看,可以直接划到最后,点击关注@IT研究僧大师兄,私信我,发送“视频截图Python脚本”,我看到后会直接发给你的。
首先,导入cv2:
import cv2
再导入os库,创建文件要用:
import os
打开视频文件,这一步是获取一个视频截图的对象。
cap = cv2.VideoCapture(video_path)
获取视频的总帧数:
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
获取视频的帧率:
fps = cap.get(cv2.CAP_PROP_FPS)
然后就能得到视频的总时长:
duration = total_frames / fps
拿到总时长的目的,就是为了后面能够根据时间来截图。
提前创建好截图保存的目录:
try:
os.mkdir(self._output_path)except OSError:
pass
然后一个比较关键的地方是计算时间点:
msec = (1000 * start_time + count * 1000 * time_interval)
其中start_time是起始时间的秒数,time_interval是时间间隔的描述,这个用来每隔多少秒进行截图。
计算得到的msec就是视频时间点的毫秒数了,然后使用这个时间设置视频的时间位置:
(1000 * self._start_time + count * 1000 * self._time_interval)
这个函数,就是把视频的拉到指定时间的位置,就和你自己在播放器上拖动视频的进度条一样。
然后读取视频的这一帧图片。
success, image = cap.read()
返回值success是bool值,true或者false,true就表示读取成功,false就是失败。
返回值image就是这个时间点的图片了。
然后就是将这个图片进行保存,存储在指定的路径中。
cv2.imwrite(out_path, image, [int(cv2.IMWRITE_JPEG_QUALITY), 100])
out_path就是要存储的路径+文件名,image是当前截取的这张图片,后面的参数是设置图片质量,这个设置100就行,最高质量。
这就完成截取图片了。
简单吗?
其实简单来说,就是分三步:
是非常简单了。这样子,只要加上相应的循环控制条件,从start_time循环到end_time,然后每一个时间间隔time_interval截取一张图,这样就能对视频按时间间隔截取批量的图片了。
这个可以想象一下,只要给出视频的小时数、分钟数、秒数,比如这种格式:
01:31:40
这就意味着要截取一小时三十一分四十秒处的图片。
然后将这个时间格式转化为毫秒数msec,然后通过
cap.set(cv2.CAP_PROP_POS_MSEC, msec)
设置视频的时间点,然后进行读取、保存就ok了。
然后通过这个就能够进行一些额外的扩展,比如获取指定时间范围内的视频截图?
只要制定开始时间start_time,截止时间end_time,然后对这个范围内按照时间间隔进行截图,就ok了。
希望能给你们启发!