使用线程安全的队列

  • 具体的方案是:https://github.com/ultralytics/ultralytics/pull/10201/commits

  • 方案:多线程,read改成grab不放入缓存内,read直接放入缓存,一个线程读取,另一个线程处理的,一个线程用作缓存,不使用opencv自己的缓存

  • 使用子码流,不使用主码流,main改成sub,减小图片的大小

  • rtsp://admin:admin12@192.168.101.1/h264/ch1/main/av_stream

  • 改成

  • rtsp://admin:admin12@192.168.101.1/h264/ch1/sub/av_stream

  • 使用 cap = cv2.VideoCapture(“rtsp://192.168.1.1:554/” , cv2.CAP_FFMPEG),而不是cap = cv2.VideoCapture(“rtsp://192.168.1.1:554/”)

  • 跨帧处理送入深度学习模型处理,不处理每一帧,隔几帧处理一帧

参考:
https://blog.csdn.net/qq_41950533/article/details/125216412
https://blog.csdn.net/darkeyers/article/details/84865363
https://blog.csdn.net/qq_33764934/article/details/103482121
https://github.com/ZLMediaKit/ZLMediaKit/issues/2018
https://blog.csdn.net/submarineas/article/details/110083906
https://blog.csdn.net/David_jiahuan/article/details/103693582
https://blog.csdn.net/qq_43348528/article/details/109337221
https://blog.csdn.net/qq_29414731/article/details/120284387

cv2.CAP_PROP_FRAME_COUNT 指定视频帧的位置
cap.set(cv2.CAP_PROP_FRAME_COUNT, location)

rtsp://admin:admin12@192.168.101.1/h264/ch1/main/av_stream 主码流
rtsp://admin:admin12@192.168.101.1/h264/ch1/sub/av_stream 子码流

修改海康视频流的配置

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开发者社区

更多推荐