简介
补充:很多人说这个跟“关键帧”没有关系,我不知道“关键帧”这个次是否是特指的,为了避免大家误解,此处改为:“指定帧”。
其实已经写过两篇关于从视频中提取“指定帧”的博客了(CV | OpenCV 提取AVI视频中“指定帧”图像(修改版) ,CV | 使用Python+OpenCV提取avi和mp4格式视频中的“指定帧”)。但是今天把代码又修改完善了一下,感觉比之前的几个版本都好(哈哈自夸中。。。),因此再次贴出利用OpenCV从视频中提取“指定帧”的Python代码。
程序
功能:自动的读取指定文件夹下的所有文件,并且按照文件类型和名称的不同,提取该文件夹下所有视频,并按照名称的不同,分类保存从视频中提取的“指定帧”。
import cv2
import os
# 从.avi 类型的视频中提取图像
def splitFrames(sourceFileName):
# 在这里把后缀接上
video_path = os.path.join('./data/video_dataset/', sourceFileName + '.avi')
outPutDirName = './data/video_dataset/video/' + sourceFileName + '/'
if not os.path.exists(outPutDirName):
#如果文件目录不存在则创建目录
os.makedirs(outPutDirName)
cap = cv2. VideoCapture(video_path) # 打开视频文件
num = 1
while True:
# success 表示是否成功,data是当前帧的图像数据;.read读取一帧图像,移动到下一帧
success, data = cap.read()
if not success:
break
# im = Image.fromarray(data, mode='RGB') # 重建图像
# im.save('C:/Users/Taozi/Desktop/2019.04.30/' +str(num)+".jpg") # 保存当前帧的静态图像
cv2.imwrite( outPutDirName +str(num)+".jpg", data)
num = num + 1
# if num % 20 == 0:
# cv2.imwrite('./Video_dataset/figures/' + str(num) + ".jpg", data)
print(num)
cap.release()
# 从.mp4 数据类型的视频中提取图像
def splitFrames_mp4(sourceFileName):
# 在这里把后缀接上
video_path = os.path.join('./data/video_dataset/', sourceFileName + '.mp4')
times = 0
# 提取视频的频率,每25帧提取一个
# frameFrequency = 25
# 输出图片到当前目录vedio文件夹下
outPutDirName = './data/video_dataset/video/' + sourceFileName + '/'
# 如果文件目录不存在则创建目录
if not os.path.exists(outPutDirName):
os.makedirs(outPutDirName)
camera = cv2.VideoCapture(video_path)
while True:
times+=1
res, image = camera.read()
if not res:
# print('not res , not image')
break
# if times%frameFrequency==0:
# cv2.imwrite(outPutDirName + str(times)+'.jpg', image)
# print(outPutDirName + str(times)+'.jpg')
cv2.imwrite(outPutDirName + str(times) + '.jpg', image)
print(times)
print('图片提取结束')
camera.release()
if __name__ == '__main__':
im_file = './data/Video_dataset/'
# for im_name in im_names:
for im_name in os.listdir(im_file):
suffix_file = os.path.splitext(im_name)[-1]
if suffix_file == '.mp4':
print('~~~~~~~~~~ 从.mp4 视频提取图像 ~~~~~~~~~~~~~~~')
sourceFileName = os.path.splitext(im_name)[0]
splitFrames_mp4(sourceFileName)
elif suffix_file == '.avi' :
print('~~~~~~~~~~ 从.avi 视频提取图像 ~~~~~~~~~~~~~~~')
sourceFileName = os.path.splitext(im_name)[0]
splitFrames(sourceFileName)