{"id":112209,"date":"2020-02-10T22:49:32","date_gmt":"2020-02-10T14:49:32","guid":{"rendered":"http:\/\/4563.org\/?p=112209"},"modified":"2020-02-10T22:49:32","modified_gmt":"2020-02-10T14:49:32","slug":"%e5%a6%82%e4%bd%95%e7%94%a8-ffmpeg-%e5%90%8c%e6%97%b6%e5%a4%84%e7%90%86%e5%a4%9a%e4%b8%aa-rtsp-%e6%b5%81%ef%bc%9f","status":"publish","type":"post","link":"http:\/\/4563.org\/?p=112209","title":{"rendered":"\u5982\u4f55\u7528 ffmpeg \u540c\u65f6\u5904\u7406\u591a\u4e2a rtsp \u6d41\uff1f"},"content":{"rendered":"<div>\n<div>\n<div>\n<h1>                  \u5982\u4f55\u7528 ffmpeg \u540c\u65f6\u5904\u7406\u591a\u4e2a rtsp \u6d41\uff1f               <\/h1>\n<p> <\/p>\n<div>\n<div> <span>\u8cc7\u6df1\u5927\u4f6c : MiLoucius <\/span>  <span><i><\/i> 51<\/span> <\/div>\n<div> <\/div>\n<\/p><\/div>\n<\/p><\/div>\n<\/p><\/div>\n<div isfirst=\"1\"> <\/p>\n<h1>\u63cf\u8ff0<\/h1>\n<p> <\/p>\n<p>\u624b\u5934\u4e0a\u6709\u51e0\u5341\u4e2a rtsp \u94fe\u63a5\uff0c\u9700\u8981\u63a5\u53d7\u6240\u6709\u7684 rtsp \u89c6\u9891\u6d41\uff0c\u5e76\u8fdb\u884c\u76f8\u5e94\u7684\u5904\u7406\u3002\u73b0\u6709\u662f\u4f7f\u7528 ffmpeg-python \u5e93\u6765\u63a5\u53d7 rtsp \u6d41\u3002 \u7531\u4e8e\u6ca1\u6709\u4ec0\u4e48\u597d\u7684\u65b9\u6cd5\uff0c\u6240\u4ee5\u53ea\u80fd\u7528 python \u7684 Process \u5bf9\u8c61\u6765\u7ba1\u7406\u3002<\/p>\n<h1>\u4ee3\u7801<\/h1>\n<p> <\/p>\n<h2>\u63a5\u53d7 RTSP \u7684\u4ee3\u7801<\/h2>\n<pre><code>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 &gt; 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) <\/code><\/pre>\n<h2>\u5904\u7406\u4ee3\u7801<\/h2>\n<pre><code>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 <\/code><\/pre>\n<h2>\u4f7f\u7528<\/h2>\n<p>\u6211\u4f1a\u521d\u59cb\u5316\u4e00\u4e2a DoSth \u8fdb\u7a0b\u5bf9\u8c61\uff0c\u5e76\u6839\u636e\u4e0d\u540c\u7684 rtsp \u94fe\u63a5\u521d\u59cb\u5316\u591a\u4e2a RTSPStream \u8fdb\u7a0b\u5bf9\u8c61\u3002\u4e24\u79cd\u8fdb\u7a0b\u5bf9\u8c61\u4e4b\u95f4\u901a\u8fc7\u8fdb\u7a0b\u5b89\u5168\u7684 queue \u8fde\u63a5\u3002<\/p>\n<h1>\u95ee\u9898<\/h1>\n<p> <\/p>\n<p>\u8fd9\u79cd\u65b9\u6cd5\uff0c\u4e0d\u592a\u4f18\u96c5\uff0c\u4f46\u6ca1\u6709 \u5728\u5904\u7406 25 \u4e2a rtsp \u6d41\u7684\u65f6\u5019\u8fd8\u80fd\u6b63\u5e38\u8fd0\u884c\uff0c\u4f46\u5f53\u6570\u91cf\u589e\u52a0\u5230 28 \u5de6\u53f3\u65f6\uff0c\u5c31\u65e0\u6cd5\u8fbe\u5230\u9884\u671f\uff1aCPU \u7206\u589e\u81f3 100%\u3001\u5185\u5b58\u65e0\u9650\u5236\u589e\u957f\u2026\u2026<\/p>\n<p>\u6240\u4ee5\uff0c\u6211\u5e94\u8be5\u7528\u4ec0\u4e48\u65b9\u6cd5\u6765\u540c\u65f6\u63a5\u53d7\u8fd9\u4e48\u591a\u4e2a RTSP \u89c6\u9891\u6d41\u5462\uff1f<\/p>\n<\/p><\/div>\n<div> <b>\u5927\u4f6c\u6709\u8a71\u8aaa<\/b> (<span>0<\/span>)        <\/div>\n<div> <\/div>\n<\/p><\/div>\n<\/p><\/div>\n<ul>\n<li>\n","protected":false},"excerpt":{"rendered":"<p>\u5982\u4f55\u7528 ffmpeg \u540c\u65f6\u5904\u7406\u591a\u4e2a&hellip;<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":[],"categories":[],"tags":[],"_links":{"self":[{"href":"http:\/\/4563.org\/index.php?rest_route=\/wp\/v2\/posts\/112209"}],"collection":[{"href":"http:\/\/4563.org\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/4563.org\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/4563.org\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/4563.org\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=112209"}],"version-history":[{"count":0,"href":"http:\/\/4563.org\/index.php?rest_route=\/wp\/v2\/posts\/112209\/revisions"}],"wp:attachment":[{"href":"http:\/\/4563.org\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=112209"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/4563.org\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=112209"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/4563.org\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=112209"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}