代码
import cv2
import os
# 按帧针对视频进行提取
def transform_frame(sourcePath, targetFolder):
# 要提取视频的文件名
sourceFileName = sourcePath
# 初始化times
times = 0
# 提取视频的频率,每25帧提取一个
frameFrequency = 25
# 输出图片到指定目录文件夹下
outPutDirName = targetFolder
# 读取时视频文件流
camera = cv2.VideoCapture(sourceFileName)
# 迭代控制保存的图片文件名 从1开始计数
imgs = []
id = 1
while True:
times += 1
res, image = camera.read()
if not res:
print('not res , not image')
break
# 判断间隔多少帧
if times % frameFrequency == 0:
id += 1
targetFile = os.path.join(outPutDirName, str(id)) + '.jpg'
cv2.imwrite(targetFile, image)
print('视频图片提取成功:', targetFile)
imgs.append(targetFile)
print('图片提取结束')
camera.release()
return imgs
# 按秒针对视频进行提取
# 提取视频中图片 按照每秒提取 间隔是视频帧率
def transform_second(sourceFolder, targetFolder):
videos = os.listdir(sourceFolder) # 返回指定路径下的文件和文件夹列表。
for video_name in videos: # 依次读取视频文件
file_name = video_name.split('.')[0] # 拆分视频文件名称 ,剔除后缀
folder_name = targetFolder + file_name # 保存图片的上级目录+对应每条视频名称 构成新的目录存放每个视频的
os.makedirs(folder_name, exist_ok=True) # 创建存放视频的对应目录
vc = cv2.VideoCapture(sourceFolder + video_name) # 读入视频文件
fps = vc.get(cv2.CAP_PROP_FPS) # 获取帧率
print(fps) # 帧率可能不是整数 需要取整
rval = vc.isOpened() # 判断视频是否打开 返回True或False
c = 1
res_videos = []
while rval: # 循环读取视频帧
rval, frame = vc.read() # videoCapture.read() 函数,第一个返回值为是否成功获取视频帧,第二个返回值为返回的视频帧:
pic_path = folder_name + '/'
if rval:
if (c % round(fps) == 0): # 每隔fps帧进行存储操作 ,可自行指定间隔
file_path = pic_path + 'video_' + str(round(c / fps)) + '.png'
cv2.imwrite(file_path,frame) # 存储为图像的命名 video_数字(第几个文件).png
print(file_path)
res_videos.append(file_path)
cv2.waitKey(1) # waitKey()--这个函数是在一个给定的时间内(单位ms)等待用户按键触发;如果用户没有按下键,则接续等待(循环)
c = c + 1
else:
break
vc.release()
print('save_success' + folder_name)
return res_videos
if __name__ == '__main__':
sourcePath = r'E:/data/video/'
targetFolder = './result/'
# datas = transform_frame(sourcePath,targetFolder)
datas = transform_second(sourcePath, targetFolder)
print(datas)
测试