视频逐帧抽取。
以每一帧的时间戳做为文件名。
多线程。
将脚本文件所在的目录下视频文件抽帧,并以“视频文件名”作为存放帧的文件夹。
import cv2
import os
import sys
import threading
import tqdm
def extract_video_frame(src_video, dest_dir):
assert os.path.splitext(src_video)[-1] == '.mp4', {src_video} + 'is not a video'
meta_info = os.stat(src_video)
ctime = meta_info.st_ctime
#ctime_ns = meta_info.st_atime_ns
print('create time: ', ctime)
#print('ctime_ns: ',ctime_ns)
#print('time: ', os.path.getctime(video_path))
cap = cv2.VideoCapture(src_video)
if os.path.exists(dest_dir) is False:
os.makedirs(dest_dir, True)
fps = cap.get(cv2.CAP_PROP_FPS)
frame_count = cap.get(cv2.CAP_PROP_FRAME_COUNT)
frame_interval = 1/fps
timestamp = ctime
assert cap.isOpened() == True, {src_video} + 'can not be opend'
for i in tqdm.trange(int(frame_count), desc=src_video):
ret, frame = cap.read()
if ret:
millseconds = cap.get(cv2.CAP_PROP_POS_MSEC)
timestamp = ctime + millseconds
name = dest_dir + '\\' + str(timestamp) + '.jpg'
#name = os.path.join(dest_dir, str(timestamp) + '.jpg')
cv2.imwrite(name, frame)
else:
print('finished!')
break
cap.release()
def run():
print('current path is: ', sys.argv[0])
print('current dir: ', os.getcwd())
print('absolute dir: ', os.path.abspath('.'))
input_dir = os.getcwd()
video_list = []
for video_name in os.listdir(input_dir):
file = os.path.join(input_dir, video_name)
if os.path.isfile(file):
if os.path.splitext(file)[-1] == '.mp4':
video_list.append(file)
print(video_list)
for video_name in video_list:
threading.Thread(target = extract_video_frame, args = (video_name, os.path.splitext(video_name)[0])).start()
if __name__ == "__main__":
run()
'''
param define
cv2.VideoCapture.get(0) CV_CAP_PROP_POS_MSEC 视频文件的当前位置(播放)以毫秒为单位
cv2.VideoCapture.get(1) CV_CAP_PROP_POS_FRAMES 基于以0开始的被捕获或解码的帧索引
cv2.VideoCapture.get(2) CV_CAP_PROP_POS_AVI_RATIO 视频文件的相对位置(播放):0=电影开始,1=影片的结尾。
cv2.VideoCapture.get(3) CV_CAP_PROP_FRAME_WIDTH 在视频流的帧的宽度
cv2.VideoCapture.get(4) CV_CAP_PROP_FRAME_HEIGHT 在视频流的帧的高度
cv2.VideoCapture.get(5) CV_CAP_PROP_FPS 帧速率
cv2.VideoCapture.get(6) CV_CAP_PROP_FOURCC 编解码的4字-字符代码
cv2.VideoCapture.get(7) CV_CAP_PROP_FRAME_COUNT 视频文件中的帧数
cv2.VideoCapture.get(8) CV_CAP_PROP_FORMAT 返回对象的格式
cv2.VideoCapture.get(9) CV_CAP_PROP_MODE 返回后端特定的值,该值指示当前捕获模式
cv2.VideoCapture.get(10) CV_CAP_PROP_BRIGHTNESS 图像的亮度(仅适用于照相机)
cv2.VideoCapture.get(11) CV_CAP_PROP_CONTRAST 图像的对比度(仅适用于照相机)
cv2.VideoCapture.get(12) CV_CAP_PROP_SATURATION 图像的饱和度(仅适用于照相机)
cv2.VideoCapture.get(13) CV_CAP_PROP_HUE 色调图像(仅适用于照相机)
cv2.VideoCapture.get(14) CV_CAP_PROP_GAIN 图像增益(仅适用于照相机)(Gain在摄影中表示白平衡提升)
cv2.VideoCapture.get(15) CV_CAP_PROP_EXPOSURE 曝光(仅适用于照相机)
cv2.VideoCapture.get(16) CV_CAP_PROP_CONVERT_RGB 指示是否应将图像转换为RGB布尔标志
cv2.VideoCapture.get(17) CV_CAP_PROP_WHITE_BALANCE × 暂时不支持
cv2.VideoCapture.get(18) CV_CAP_PROP_RECTIFICATION 立体摄像机的矫正标注(目前只有DC1394 v.2.x后端支持这个功能)
'''