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()



以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持中文源码网。