之前需要做一些目标检测的训练,需要自己采集一些数据集,写了一个小demo来实现图片的采集
使用方法:
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()
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持武林网。
新闻热点
疑难解答
图片精选