python视频与帧图片的相互转化,以及查看视频分辨率

1.拆分视频为帧图片

import cv2
def video2frame(videos_path,frames_save_path,time_interval):
    vidcap = cv2.VideoCapture(videos_path)
    success, image = vidcap.read()
    count = 0
    while success:
        success, image = vidcap.read()
    count = count + 1
    if count % time_interval == 0:
        cv2.imencode('.jpg', image)[1].tofile(frames_save_path + "frame%d.jpg" % count)
    print('一共{}张，选取其中{}张'.format(count,int(count/time_interval)))
 
if __name__ == '__main__':
    videos_path = 'C:\\Users\\cjk\\Desktop\\test.mp4' #视频的存放路径
    frames_save_path = 'C:\\Users\\cjk\\Desktop\\images\\' #视频切分成帧之后图片的保存路径
    time_interval = 5 #每5帧保存一次
    video2frame(videos_path, frames_save_path, time_interval)

2.用帧图片合成视频

import cv2
import os
import numpy as np
from PIL import Image
 
def frame2video(im_dir,video_dir,fps):
    im_list = os.listdir(im_dir)
    im_list.sort(key=lambda x: int(x.replace("frame","").split('.')[0]))
    img = Image.open(os.path.join(im_dir,im_list[0]))
    img_size = img.size #获得图片分辨率，im_dir文件夹下的图片分辨率需要一致
    fourcc = cv2.VideoWriter_fourcc(*'XVID') 
    videoWriter = cv2.VideoWriter(video_dir, fourcc, fps, img_size)
    for i in im_list:
        im_name = os.path.join(im_dir+i)
        frame = cv2.imdecode(np.fromfile(im_name, dtype=np.uint8), -1)
        videoWriter.write(frame)
    videoWriter.release()
 
if __name__ == '__main__':
    im_dir = 'C:\\Users\\cjk\\Desktop\\images\\'#帧存放路径
    video_dir = 'C:\\Users\\cjk\\Desktop\\ok.avi' #合成视频存放的路径
    fps = 30 #帧率
    frame2video(im_dir, video_dir, fps)

3.查看视频分辨率

import os,time
import cv2

#获取文件夹中的文件名列表
def get_file_list(file_path):
    dir_list = os.listdir(file_path)
    if not dir_list:
        return
    else:
        dir_list = sorted(dir_list,key=lambda x: os.path.getmtime(os.path.join(file_path, x)))
        return dir_list

file_path = 'H:\视频\\'

dir_list = get_file_list(file_path)
#获取使用cv2获得视频的分辨率width和height。
def get_vedio_height_width(filename):
    cap = cv2.VideoCapture(filename)
    width = cap.get(cv2.CAP_PROP_FRAME_WIDTH )
    height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
    #print(width,height)
    return width,height

if __name__ == '__main__': 
    for dir in dir_list:
        width,height = get_vedio_height_width(file_path+dir)
        print('文件名:',dir)
        print('分辨率:',width,height)