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)