python一帧一帧读取视频_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