add IP camera reading

Author: chenxinfeng

README.md

@@ -65,6 +65,8 @@ cap = ffmpegcv.VideoCaptureCAM(0)
 cap = ffmpegcv.VideoCaptureCAM("Integrated Camera")
 ```
 
+## Cross platform
+
 ## GPU Accelation
 - Support **NVIDIA** card only, test in x86_64 only.
 - Works in **Windows**, **Linux** and **Anaconda**.

README_CN.md

@@ -14,6 +14,7 @@ ffmpegcv提供了基于ffmpeg的视频读取器和视频编写器，比cv2更快
 - ffmpegcv可以使用**GPU加速**编码和解码。
 - ffmpegcv支持比open-cv更多的**视频编码器**。
 - ffmpegcv原生支持**RGB**/BGR/灰度像素格式。
+- ffmpegcv支持网络**流视频读取** (网线监控相机)。
 - ffmpegcv支持ROI（感兴趣区域）操作，可以对ROI进行**裁剪**、**调整大小**和**填充**。
 总之，ffmpegcv与opencv的API非常相似。但它具有更多的编码器，并且不需要安装opencv。
 
@@ -24,6 +25,7 @@ ffmpegcv提供了基于ffmpeg的视频读取器和视频编写器，比cv2更快
 - `VideoCaptureQSV`: 使用Intel集成显卡读取视频文件.
 - `VideoCaptureCAM`：读取摄像头。
 - `VideoCaptureStream`：读取RTP/RTSP/RTMP/HTTP流。
+- `VideoCaptureStreamRT`: 读取RTSP流 (网线监控相机)，实时、低延迟。
 - `noblock`：在后台读取视频文件（更快）,使用多进程。
 
 ## 安装
@@ -293,7 +295,7 @@ cap = ffmpegcv.VideoCaptureCAM('FaceTime HD Camera', camsize_wh=(1280,720), camf
 
 2. 在Linux上VideoCaptureCAM无法列出FPS，因为`ffmpeg`无法使用Linux本机的`v4l2`模块查询设备的FPS。不过，让FPS为空也没问题。
 
-## 流读取器 （直播流，IP摄像头）
+## 流读取器 （直播流，网络监控摄像头）
 **实验性功能**。ffmpegcv提供了流读取器，与VideoFile读取器一致，更类似于相机。
 
 - 支持`RTSP`、`RTP`、`RTMP`、`HTTP`、`HTTPS`流。
@@ -325,6 +327,18 @@ while True:
     if not ret:
         break
     pass
+
+# ffmpegcv, 网络监控摄像头
+# 例如 海康威视, `101` 主视频流, `102` 子视频流
+stream_url = 'rtsp://admin:PASSWD@192.168.1.xxx:8554/Streaming/Channels/102'
+cap = ffmpegcv.VideoCaptureStreamRT(stream_url)                 # 低延迟 & 缓存
+cap = ffmpegcv.ReadLiveLast(ffmpegcv.VideoCaptureStreamRT, stream_url) #不缓存
+while True:
+    ret, frame = cap.read()
+    time.sleep(1/30)
+    if not ret:
+        break
+    pass
 ```
 
 ## FFmpegReaderNoblock

ffmpegcv/__init__.py

@@ -6,7 +6,7 @@
 from .ffmpeg_writer_stream_realtime import FFmpegWriterStreamRT
 from .ffmpeg_reader_qsv import FFmpegReaderQSV
 from .ffmpeg_writer_qsv import FFmpegWriterQSV
-from .ffmpeg_noblock import noblock
+from .ffmpeg_noblock import noblock, ReadLiveLast
 from .video_info import get_num_NVIDIA_GPUs
 import shutil
 from subprocess import DEVNULL, check_output

ffmpegcv/ffmpeg_reader_camera.py

@@ -205,16 +205,15 @@ def __init__(self, vid, q):
         self.q = q
 
     def run(self):
+        q = self.q
         while True:
             if not self.vid.isOpened():
                 break
             ret, img = self.vid.read_()
 
-            try:
-                self.q.put_nowait((ret, img))  # drop frames
-            except Exception:
-                pass
-            continue
+            if q.full():
+                q.get() # drop frames
+            q.put((ret, img))  
 
 
 class FFmpegReaderCAM:

ffmpegcv/ffmpeg_reader_noblock.py

@@ -28,7 +28,7 @@ def __init__(self,
         self.shared_array = shared_array
         self.vcap_args = vcap_args
         self.vcap_kwargs = vcap_kwargs
-        self.q = Queue(maxsize=(NFRAME-1)*2)
+        self.q = Queue(maxsize=(NFRAME-2)*2)
         self.vcap_fun = vcap_fun
         self.has_init = False
         self.process = None

ffmpegcv/ffmpeg_reader_stream_realtime.py

@@ -34,9 +34,10 @@ def VideoReader(
                 resize_keepratio, resize_keepratioalign)
         vid.size = (vid.width, vid.height)
 
+        rtsp_opt = '-rtsp_transport tcp ' if stream_url.startswith('rtsp://') else ''
         vid.ffmpeg_cmd = (
             f"ffmpeg -loglevel warning "
-            '-rtsp_transport tcp '
+            f' {rtsp_opt} '
             '-fflags nobuffer -flags low_delay -strict experimental '
             f' -vcodec {vid.codec} -i {stream_url}'
             f" {filteropt} -pix_fmt {pix_fmt}  -f rawvideo pipe:"

ffmpegcv/version.py

@@ -1 +1 @@
-__version__='0.3.8'
+__version__='0.3.9'