MJ : save an img from rtsp stream.

rtspToImage.py

@@ -0,0 +1,48 @@
+# coding:utf-8
+
+import cv2
+import dlib
+import time
+import glob
+
+cnt = 0
+def getImgFromCamera(folder_path):
+    detector = dlib.get_frontal_face_detector()
+    #cap = cv2.VideoCapture("rtsp://192.168.169.30:8554/")#获取网络摄像机
+    cap = cv2.VideoCapture(0)#/dev/video0
+    print("cap.isOpened:",cap.isOpened())
+
+    #while i<10:
+    while(cap.isOpened()):
+        if cap.isOpened() == True:
+            ret, frame = cap.read()#读一帧
+            
+            # #探测图片中的人脸
+            # gray_img = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+            # dets = detector(gray_img, 1)
+            # print("人脸数：", len(dets))
+            
+            # #绘制矩形框出人脸区域
+            # for e, d in enumerate(dets):
+                # x1 = d.top() if d.top() > 0 else 0
+                # y1 = d.bottom() if d.bottom() > 0 else 0
+                # x2 = d.left() if d.left() > 0 else 0
+                # y2 = d.right() if d.right() > 0 else 0 
+                
+                # cv2.rectangle(frame, (x2, x1), (y2, y1), (0, 255, 0), 2)
+                
+            # 当发现人脸进行操作:显示,保存图片文件,保存为视频,推送rtmp
+            cv2.imshow("capture", frame)#显示一帧图像
+            cv2.imwrite(folder_path + str(cnt) + '.jpg', frame)# 保存图片文件
+                        
+            
+            if cv2.waitKey(1) & 0xFF == ord('q'):
+                break
+        
+    cap.release()
+    cv2.destroyAllWindows()
+ 
+# 测试
+if __name__ == '__main__':
+    folder_path = '/home/mj/experiment/golang/src/cameraToRtmp/'
+    getImgFromCamera(folder_path)