描述

手头上有几十个 rtsp 链接，需要接受所有的 rtsp 视频流，并进行相应的处理。现有是使用 ffmpeg-python 库来接受 rtsp 流。 由于没有什么好的方法，所以只能用 python 的 Process 对象来管理。

代码

接受 RTSP 的代码

class RTSPStream(Process):     def __init__(self, queue, stream_id, stream_url):         Process.__init__(self)         self.q = q         self.stream_id = stream_id         self.stream_url = stream_url      def get_hnw(self, url):         probe = ffmpeg.probe(url, rtsp_transport='tcp')         video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None)         width = video_stream['width']         height = video_stream['height']         return height, width      def run(self):             h, w = get_hnw(self.stream_url)             assert h != 0             assert w != 0             out = (                 ffmpeg                 .input(url, rtsp_transport='tcp')                 .output('pipe:', format='rawvideo', pix_fmt='bgr24', loglevel="quiet", r=1)                 .run_async(pipe_stdout=True)             )             cnt_empty = 0             while True:                 in_bytes = out.stdout.read(h*w*3)                 if not in_bytes:                     cnt_empty += 1                     if cnt_empty > 10:                         break                     sleep(0.1)                     continue                     cnt_empty = 0                 frame = np.frombuffer(in_bytes, dtype=np.uint8).reshape(h, w, 3)                 data = {                     'stream_id': self.stream_id,                     'origin': frame                 }                 if not queue.full():                     self.queue.put(data) 

处理代码

class DoSth(Process):     def __init__(self, queue):         Process.__init__(self)         self.queue = queue      def run(self):         while True:             if queue.empty():                 sleep(0.05)                 continue             data = queue.get()             # do sth to ndarray 

使用

我会初始化一个 DoSth 进程对象，并根据不同的 rtsp 链接初始化多个 RTSPStream 进程对象。两种进程对象之间通过进程安全的 queue 连接。

问题

这种方法，不太优雅，但没有 在处理 25 个 rtsp 流的时候还能正常运行，但当数量增加到 28 左右时，就无法达到预期：CPU 爆增至 100%、内存无限制增长……

所以，我应该用什么方法来同时接受这么多个 RTSP 视频流呢？

大佬有話說 (0)