import cv2
# 人脸数据采集
def datasets():
print('准备开始人脸数据采集')
mydict = {}
while True:
print("输入'q'停止添加")
face_id = input('请设置新的人脸id(id为数字):')
if face_id == 'q':
break
face_name = input('请输入新的人脸name(name为英文或字母):')
if face_name == 'q':
break
mydict[face_id] = face_name
# print(mydict)
count = 0
vid_cam = cv2.VideoCapture(0)
face_detector = cv2.CascadeClassifier('./data/haarcascade_frontalface_default.xml')
while True:
_, image_frame = vid_cam.read()
gray = cv2.cvtColor(image_frame, cv2.COLOR_BGR2GRAY)
faces = face_detector.detectMultiScale(gray, 1.3, 5)
for (x, y, w, h) in faces:
cv2.rectangle(image_frame, (x, y), (x + w, y + h), (255, 0, 0), 2)
count += 1
cv2.imwrite("./dataset/User." + str(face_id) + '.' + str(count) + ".jpg", gray[y:y + h, x:x + w])
cv2.imshow('frame', image_frame)
if cv2.waitKey(100) & 0xFF == ord('q'):
break
elif count > 60:
print('%s:人脸数据采集完成!' % (mydict[face_id]))
break
vid_cam.release()
cv2.destroyAllWindows()
print(mydict)
return mydict

蔡军帅
- 粉丝: 610
最新资源
- NeilYSJ-Advanced-Algorithm-Design-and-Analysis-30340-1747168797789.zip
- 网络通信及安全事业部组织结构图(1).doc
- JAVA语言程序设计期末考试试题及答案.doc
- 国家CAD高级绘图师考试试卷模拟.doc
- shgopher-408-32820-1747172783865.zip
- 企业计算机网络维护策划方案(1).doc
- 键盘设计与制造-3D打印与PCB电路板设计-瀚文键盘Dynamic模块复刻项目包含05Ctrl拓展板09OLED拓展板10Encoder拓展板及Type-C转换小板-为机械键盘爱好.zip
- 东北农业大学2021年9月《电子商务》技术基础作业考核试题及答案参考2.docx
- 医药公司网站建设方案(1).doc
- YD_T_2329.2-2011_分组通信数据网(PTDN)体系架构_第2部分_链路层(1).pdf
- 《计算机程序设计基础》教学课件:考试说明及相关题型(1).ppt
- YD_T_3106-2016_电信和互联网服务_用户个人信息保护技术要求_移动应用商店(1).pdf
- 基于单片机的温度警报系统课程设计论文大学论文(1).doc
- 北华航天工业学院软件工程练习题及答案.doc
- 第8周Windows网络的安装与配置DNS服务器的安装与配置说课讲解.ppt
- 中软国际JAVA校园招聘笔试真题(1).pdf
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈


