python提取視頻第一幀圖片

一.實現代碼

# -*- coding: utf-8 -*-
import cv2
from PIL import Image
from io import BytesIO

def tryTime(maxTry, timeout=random.random()):
    """
    重試
    :param maxTry:重試次數
    :param timeout:睡眠時間
    :return:
    """
    def wrap1(func):
        #functools.wraps 可以將原函數對象的指定屬性復制給包裝函數對象,
        @functools.wraps(func)
        def __decorator(*args,**kwargs):
            tryTime = 0
            while tryTime < maxTry:
                try:
                    return func(*args,**kwargs)
                except Exception:
                    pass
                    # print("TryTime_except==%s"%traceback.format_exc())
                    # logDebug("tryTime","TryTime_except==%s"%traceback.format_exc())
                tryTime += 1
                time.sleep(timeout)
        return __decorator
    return wrap1

@tryTime(2)
def get_video_cover(url):
    cap = cv2.VideoCapture(url)
    rate = cap.get(5)
    frame_number = cap.get(7)  # 視頻文件的幀數
    if rate==0:
        duration=0
    else:
        duration = int(frame_number / rate)  # 單位秒
    cap.set(1, 1)  # 取它的第一幀
    rval, frame = cap.read()  # 如果rval為False表示這個視頻有問題，為True則正常
    data = cv2.imencode(".jpg", frame)[1].tobytes()  # 將圖片轉為jpg格式的二進制流
    cap.release()
    height = frame.shape[0]  # 高度
    width = frame.shape[1]  # 寬度
    return data, duration, width, height


def get_image_size(data):
    im = Image.open(BytesIO(data))
    return im.size