python视频转图片

python视频转图片

import cv2
import os

def video2imgs(videoPath, imgPath):
    if not os.path.exists(imgPath):
        os.makedirs(imgPath)             # 目标文件夹不存在，则创建
    cap = cv2.VideoCapture(videoPath)    # 获取视频
    judge = cap.isOpened()                 # 判断是否能打开成功
    print(judge)
    fps = cap.get(cv2.CAP_PROP_FPS)      # 帧率，视频每秒展示多少张图片
    print('fps:',fps)

    frames = 1                           # 用于统计所有帧数
    count = 1                            # 用于统计保存的图片数量

    while(judge):
        flag, frame = cap.read()         # 读取每一张图片 flag表示是否读取成功，frame是图片
        if not flag:
            print(flag)
            print("Process finished!")
            break
        else:
            if frames % 10 == 0:         # 每隔10帧抽一张
                imgname = 'jpgs_' + str(count).rjust(3,'0') + ".jpg"
                newPath = imgPath + imgname
                print(imgname)
                cv2.imwrite(newPath, frame, [cv2.IMWRITE_JPEG_QUALITY, 100])
                # cv2.imencode('.jpg', frame)[1].tofile(newPath)
                count += 1
        frames += 1
    cap.release()
    print("共有 %d 张图片"%(count-1))
video2imgs('/home/jim/Documents/document_fly/10_9/111.webm','./jpgs/')

python图片转视频

import cv2
import torch
import numpy as np
import os

path = './jpgs'
img = cv2.imread(os.path.join(path,'jpgs_001.jpg'))
# 获取图片尺寸
imgInfo = img.shape
size = (imgInfo[1],imgInfo[0])
print(size)
filelist = os.listdir(path)
filelist.sort()
fps = 4  # 视频每秒组成的原始帧数，由于之前是抽帧提取图片的，我想在这里设小一点
fourcc = cv2.VideoWriter_fourcc(*'XVID')				# 设置视频编码格式
video = cv2.VideoWriter('./2.avi', fourcc, fps, size)
print('a')
# 视频保存在当前目录下
for item in filelist:
    if item.endswith('.jpg') or item.endswith('.JPG'):
        print(item)
        item = os.path.join(path,item)
        # 路径中若存在为中文名
        # img = cv2.imdecode(np.fromfile(item, dtype=np.uint8), 1)
        # 路径为英文名
        img = cv2.imread(item)
        video.write(img)

video.release()