python-opencv文件夹中所有视频按顺序截图片并按顺序命名

使用背景

有时候我们需要把文件夹中的视频按照一定的帧率截取图片,如一秒取三张,为了实现这一需求,我特地编写了代码实现,并且附上了十分详细的说明,为了方便大家改代码实现自己需求(主要为了照顾刚学python的或者只需要用这一需求的小伙伴),本博文的特点如下:

  • 直接改变输入文件夹和输出文件夹的位置,就可以实现功能
  • 输出的图片能按照一定格式命名,这里是以20210823_0001命名
  • 逐一遍历文件夹中视频,序号之间可以连续(也可不连续,需要改代码)
  • 利用双线性插值方法,就算图片变大了,也能保证分辨率(关于双线性,这里只是调用opencv方法实现,具体实现代码和原理可看我下一篇博文)
  • 代码解释十分详细,一看就懂

十分详细代码实现

首先先说明需要自己修改的参数,代码如下:

filepath = 'C:/Users/ZFG/Desktop/1111' #视频文件夹所在目录
data='20210823'   #要是储存的文件为20210823_0001格式,则为前半部分
save_filename='C:/Users/ZFG/Desktop/2222/'  #储存图片的文件夹的地址
timeF = 6   #根据一秒取多少帧设置,比如我的视频是24帧/秒,取6则一秒取三张
kernal=(700,700)  #设置输出的大小,根据自己需求设置

然后再设置一个方法,来储存截取后的图片:

def saveImage(image,SaveAddress,num):  #image为读取的图片,SaveAddress为需要存的地址,num为截取图片时候记录的序号
    address= SaveAddress+data+'_'+str(num).zfill(4)+'.jpg'   #这里设置输出格式
    cv2.imwrite(address,image) #这里为存图片

之后读取文件夹:

pathDir = os.listdir(filepath)
i=0
j=0
for allDir in pathDir:  #遍历文件夹中的每一个视频
    videopath =filepath+'/'+ allDir
    videoCapture=cv2.VideoCapture(videopath)  #输入绝对路径
    untill,picture=videoCapture.read()  #读取视频,视频读取完的时候,返回的untill为False,表示视频读取完毕
    while untill:
        i+=1
        if (i%timeF==0):    #根据原视频的帧率看截图图片的频率
            j+=1
            picture=cv2.resize(picture,kernal,cv2.INTER_LINEAR)  #这里调用了opencv中的双线性插值法,要是图片增加很快,保证了图片精度
            saveImage(picture,save_filename,j)  #调用我们之前描述的方法
            pass
        untill, picture = videoCapture.read()  #再次看视频是否结束,结束了则until为False

将上诉代码结合后,最终程序如下所示:

import cv2
import os

i = 0
j = 0
pathDir = os.listdir(filepath)
filepath = 'C:/Users/ZFG/Desktop/1111'
data='20210823'
save_filename='C:/Users/ZFG/Desktop/2222/'
timeF = 6
kernal=(700,700)
def saveImage(image,SaveAddress,num):
    address= SaveAddress+data+'_'+str(num).zfill(4)+'.jpg'
    cv2.imwrite(address,image)
for allDir in pathDir:
    videopath =filepath+'/'+ allDir
    videoCapture=cv2.VideoCapture(videopath)
    untill,picture=videoCapture.read()
    while untill:
        i+=1
        if (i%timeF==0):
            j+=1
            picture=cv2.resize(picture,kernal,cv2.INTER_LINEAR)
            saveImage(picture,save_filename,j)
            pass
        untill, picture = videoCapture.read()

最后看下效果吧:
第一张图是文件夹中的视频,帧率是24帧/秒,第二张图是一秒取三张图片后,图片储存在文件夹中的图。

猜你喜欢

转载自blog.csdn.net/weixin_50233398/article/details/119955643