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

﹏ヽ暗。殇╰゛Y 2023-03-04 09:28 13阅读 0赞

## 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)