python調用攝像頭拍攝數據集

這篇文章主要爲大家詳細介紹了Python調用攝像頭拍攝數據集，具有一定的參考價值，感興趣的小夥伴們可以參考一下

之前需要做一些目標檢測的訓練，需要自己採集一些數據集，寫了一個小demo來實現圖片的採集

使用方法：

• 指定name的名稱，name爲分類的標籤
• 按n鍵拍攝圖片
• 程序會在當前目錄下生成一個pictures的文件夾，圖片存放在其中
  

print("正在初始化攝像頭...")
import cv2
import os
import datetime
cap = cv2.VideoCapture(0)
print("初始化成功！")
 
# name='play_phone'
# name='haqian'
# name='spleeing'
# name='zhengchang'
# name="zhedang"
name="waitou"
 
savedpath =r'./pictures/'+name
isExists = os.path.exists(savedpath)
if not isExists:
  os.makedirs(savedpath)
  print('path of %s is build' % (savedpath))
else:
  print('path of %s already exist and rebuild' % (savedpath))
print("按N鍵拍攝圖片")
i=0
while(True):
  ret, frame = cap.read()
  gray = cv2.cvtColor(frame, 1)
  cv2.imshow('test',frame)
  now = datetime.datetime.now()
  now = now.strftime('%m-%d-%H-%M-%S')
  savedname = '/'+name+ '_' + str(i) + '_{0}' '.jpg'.format(now)
  if cv2.waitKey(1) & 0xFF == ord('n'):  #按N拍攝
    i += 1
    cv2.imwrite(savedpath + savedname, frame)
    cv2.namedWindow("Image")
    cv2.imshow("Image", frame)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
 
cap.release()
cv2.destroyAllWindows()

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持神馬文庫。