python调用摄像头拍摄数据集
之前需要做一些目标检测的训练，需要自己采集一些数据集，写了一个小demo来实现图片的采集
使用方法：
- 指定name的名称，name为分类的标签
- 按n键拍摄图片
- 程序会在当前目录下生成一个pictures的文件夹，图片存放在其中
print("正在初始化摄像头...")
importcv2
importos
importdatetime
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)
ifnotisExists:
os.makedirs(savedpath)
print('pathof%sisbuild'%(savedpath))
else:
print('pathof%salreadyexistandrebuild'%(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)
ifcv2.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()
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。