opencv在读取视频的时候,会将提前读取的视频帧放到缓存中,但是缓存的大小是有限的,所以当我们在对视频帧做长时间的处理时,会出现缓存被覆写的情况,这个时候capture.read()函数就拿不到下一个标号的视频帧了,缓存内的视频帧被覆写了,导致了视频帧的丢失

.read()函数拿不到指定标号的视频帧时,会报错,而且报错内容比较多

[hevc] Could not find ref with POC or GStreamer

[h264 @ ] error while decoding MB 15 51, bytestream -31

[h264 @ ] non-existing PPS 0 referenced

[h264 @ ] decode_slice_header error

[h264 @ ] no frame!

一般写入视频前,会对视频帧做长时间的处理,若是处理太长时间,会导致opencv的缓存被用完,然后read拿不到指定的视频帧

常见的方案是:多线程, 一个线程读取,一个处理的,另外一个写入;可能需要使用线程安全的队列,但是随着时间的流逝,一般队列的容量会越来越大,内存会被耗尽

所以需要每隔几帧处理一帧,不处理每一个帧,来不及处理的

或者每隔几帧丢弃1帧,这样保证了队列的大小不会持续变大

grab, retrieve和read

opencv的read函数,是将视频标号+1,然后将缓存中的视频帧读取到指定的内存中,若是标号所在的视频帧丢失了或者被覆写了,就会报错的

grab函数只负责将视频标号+1,retrieve函数负责将缓存中的视频帧读取到指定的内存中

所以更推荐使用grab, retrieve函数的

下面的codes就是每隔几帧获取一帧然后处理的,不处理每一帧,来不及

读取和写入视频的codes

import cv2
import time
from datetime import datetime

def openvideo():
    path = r'rtsp://administer:123456789@192.168.100.100/h265/ch1/sub/av_stream'
    vid = cv2.VideoCapture(path)
    ret = True
    print(cv2.getBuildInformation())
    outpath = '/home/Desktop/scripts/kkk.avi'
    fps = vid.get(cv2.CAP_PROP_FPS)
    width = int(vid.get(cv2.CAP_PROP_FRAME_WIDTH))
    height = int(vid.get(cv2.CAP_PROP_FRAME_HEIGHT))
    sizes = (width, height)

    cnt = 0
    interval = 6
    exist_path = r'/home/jiahan/Documents/AIMining/SceneVideo/1/images'

    videowriter = cv2.VideoWriter(outpath, cv2.VideoWriter_fourcc(*'XVID'), fps, sizes)

    while ret:
        ret = vid.grab()  # go to next pointer, no read to memery
        cnt += 1
        ############## vid.read() # go to next pointer and read to memery
        if cnt % interval == 0:
            ret, frame = vid.retrieve() # read to memery
            imgPath = str(cnt) + ".jpg"
            cv2.imwrite(imgPath, frame)
            videowriter.write(frame)
            # wait
            # grab drop some frame
    videowriter.release()

if __name__=="__main__":
    openvideo()

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐