mirror of
https://github.com/mijian1988/cameraToRtmp.git
synced 2025-09-27 01:35:52 +08:00
49 lines
1.6 KiB
Python
Executable File
49 lines
1.6 KiB
Python
Executable File
# 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)
|