提取avi视频中的图片

import cv2    #cv2对应的库名是opencv-python
#import traceback
vidcap = cv2.VideoCapture('video/video.avi')  #avi文件存放路径,使用源文件需要确定是否损坏
success,image = vidcap.read()
count = 0
success = True      

while success:##success:         #循环处理True
    success,image = vidcap.read()   
    try:        #使用try判断                
        cv2.imwrite("pic1/frame%d.jpg" % count, image)  # save frame as JPEG file
    except:   #如果异常
        print(count)     #报出异常信息
        continue          #这里因为我的avi文件710后面的侦已经都损坏了,这个场景后面的数据都是损坏的,所以continue和break两个是都可以使用的,如果后面有数据的话用break的话,后面的数据就中断处理啦,严谨些用continue会好些那。  
    if cv2.waitKey(10) == 27:  #按exit键退出循环。
        break
    count += 1
    # except Exception as e:
    #     continue

将武汉加油写入图片中

from PIL import Image, ImageDraw, ImageFont
import cv2
import os


def draw(pic):
    img = cv2.imread('pic/' + pic)
    img = img[:, :, (2, 1, 0)]

    blank = Image.new("RGB", [len(img[0]), len(img)], "white")
    drawObj = ImageDraw.Draw(blank)

    n = 10

    font = ImageFont.truetype('C:/Windows/Fonts/Microsoft YaHei UI/msyhbd.ttc', size=n - 1)

    for i in range(0, len(img), n):
        for j in range(0, len(img[i]), n):
            text = '武汉加油'
            drawObj.ink = img[i][j][0] + img[i][j][1] * 256 + img[i][j][2] * 256 * 256
            drawObj.text([j, i], text[int(j / n) % len(text)], font=font)
            print('完成处理——', i, j)

    blank.save('new/new_' + pic, 'jpeg')


filelist = os.listdir('pic')
for file in filelist:
    draw(file)

将所有处理过的图片制作出mp4视频。

import time
import os
import cv2
import re


def resort(list):
    for i in range(len(list)-1):
        for j in range(len(list)-1):
            if int(re.findall(r'\d+', list[j])[0]) > int(re.findall(r'\d+', list[j+1])[0]):
                list[j], list[j+1] = list[j+1], list[j]
    return list


def picvideo(path, size):
    # path = r'C:\Users\Administrator\Desktop\1\huaixiao\\'#文件路径
    filelist = os.listdir(path)  # 获取该目录下的所有文件名
    filelist = resort(filelist)


    '''
    fps:
    帧率:1秒钟有n张图片写进去[控制一张图片停留5秒钟,那就是帧率为1,重复播放这张图片5次]
    如果文件夹下有50张 534*300的图片,这里设置1秒钟播放5张,那么这个视频的时长就是10秒
    '''
    fps = 24
    # size = (591,705) #图片的分辨率片
    file_path = 'video/new.mp4'  # 导出路径
    fourcc = cv2.VideoWriter_fourcc('D', 'I', 'V', 'X')  # 不同视频编码对应不同视频格式(例:'I','4','2','0' 对应avi格式)

    video = cv2.VideoWriter(file_path, fourcc, fps, size)

    for item in filelist:
        if item.endswith('.jpg'):  # 判断图片后缀是否是.png
            item = path + '/' + item
            img = cv2.imread(item)  # 使用opencv读取图像,直接返回numpy.ndarray 对象,通道顺序为BGR ,注意是BGR,通道值默认范围0-255。
            video.write(img)  # 把图片写进视频

    video.release()  # 释放


picvideo(r'new', (960, 544))

以上代码已经实现,不过还要花点时间消化下。

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐