看到有按照帧提取视频中的图片的例子,但是个人只需用到每秒的图片。
以下是用到的代码:
import os
import cv2
def save_img2(): # 提取视频中图片 按照每秒提取 间隔是视频帧率
video_path = r'C:/Users/Louise喵/Desktop/video1/' # 视频所在的路径
f_save_path = 'D:/image/' # 保存图片的上级目录
videos = os.listdir(video_path) # 返回指定路径下的文件和文件夹列表。
for video_name in videos: # 依次读取视频文件
file_name = video_name.split('.')[0] # 拆分视频文件名称 ,剔除后缀
folder_name = f_save_path + file_name # 保存图片的上级目录+对应每条视频名称 构成新的目录存放每个视频的
os.makedirs(folder_name, exist_ok=True) # 创建存放视频的对应目录
vc = cv2.VideoCapture(video_path + video_name) # 读入视频文件
fps = vc.get(cv2.CAP_PROP_FPS) # 获取帧率
print(fps) # 帧率可能不是整数 需要取整
rval = vc.isOpened() # 判断视频是否打开 返回True或False
c = 1
while rval: # 循环读取视频帧
rval, frame = vc.read() # videoCapture.read() 函数,第一个返回值为是否成功获取视频帧,第二个返回值为返回的视频帧:
pic_path = folder_name + '/'
if rval:
if (c % round(fps) == 0): # 每隔fps帧进行存储操作 ,可自行指定间隔
cv2.imwrite(pic_path + 'video_' + str(round(c/fps)) + '.png', frame) #存储为图像的命名 video_数字(第几个文件).png
print('video_' + str(round(c/fps)) + '.png')
cv2.waitKey(1) # waitKey()--这个函数是在一个给定的时间内(单位ms)等待用户按键触发;如果用户没有按下键,则接续等待(循环)
c = c + 1
else:
break
vc.release()
print('save_success' + folder_name)
save_img2()
有几点需要注意:
- 提取出的图片位置不要放在C盘,放在C盘会显示文件夹为空,涉及到权限问题没有将图片存入文件夹中。
- 逻辑上是读取视频帧率,然后根据FPS来指定间隔进行存储图片,从而达到每秒存储一次图片。
- 以上代码可以批量读取多个视频,前提是在同一个路径下。一个视频对应的图集均以视频名存于指定位置。
- 视频存储的路径下最好只有需要处理的视频,以防报错。
以下是一个读取一个视频后的显示,最后显示成功存储在指定路径下。
文件夹显示如下:
提取成功。