首页 > 编程 > Python > 正文

python调用摄像头拍摄数据集

2019-11-25 12:49:34
字体:
来源:转载
供稿:网友

之前需要做一些目标检测的训练,需要自己采集一些数据集,写了一个小demo来实现图片的采集

使用方法:

  • 指定name的名称,name为分类的标签
  • 按n键拍摄图片
  • 程序会在当前目录下生成一个pictures的文件夹,图片存放在其中
print("正在初始化摄像头...")import cv2import osimport datetimecap = cv2.VideoCapture(0)print("初始化成功!") # name='play_phone'# name='haqian'# name='spleeing'# name='zhengchang'# name="zhedang"name="waitou" savedpath =r'./pictures/'+nameisExists = 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=0while(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()

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

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表