python 从本地摄像头和网络摄像头截取图片

import cv2

# 获取本地摄像头

# folder_path 截取图片的存储目录

def get_img_from_camera_local(folder_path):

cap = cv2.VideoCapture(0)

i = 1

while True:

ret, frame = cap.read()

cv2.imshow("capture", frame)

print str(i)

cv2.imwrite(folder_path + str(i) + '.jpg', frame)  # 存储为图像

if cv2.waitKey(1) & 0xFF == ord('q'):

break

i += 1

cap.release()

cv2.destroyAllWindows()

# 获取网络摄像头，格式：rtsp://username:[email protected]/

# folder_path 截取图片的存储目录

def get_img_from_camera_net(folder_path):

cap = cv2.VideoCapture('rtsp://username:[email protected]/')

i = 1

while True:

ret, frame = cap.read()

cv2.imshow("capture", frame)

print str(i)

cv2.imwrite(folder_path + str(i) + '.jpg', frame)  # 存储为图像

if cv2.waitKey(1) & 0xFF == ord('q'):

break

i += 1

cap.release()

cv2.destroyAllWindows()

# 测试

if __name__ == '__main__':

folder_path = 'D:\\img_from_camera\\'

get_img_from_camera_local(folder_path)