PaddleHub人脸检测实现一键查人数
收藏
想必各位同学都经历过老师上课点名的场景,想逃不能逃才最寂寞……
如今,老师们也可是用起了人工智能,仅需拍一张照片,就能一键查出今天来了多少人,是不是很便捷?怎么实现呢?其实也很简单了,下面做简要介绍:
一、PaddleHub简介
便捷地获取PaddlePaddle生态下的预训练模型,完成模型的管理和一键预测。配合使用Fine-tune API,可以基于大规模预训练模型快速完成迁移学习,让预训练模型能更好地服务于用户特定场景的应用。
更多模型获取请查询官网:https://www.paddlepaddle.org.cn/hub
二、实现过程
采用PaddleHub提供的人脸检测模型Ultra-Light-Fast-Generic-Face-Detector-1MB,完成人脸检测。该模型是针对边缘计算设备或低算力设备(如用ARM推理)设计的实时超轻量级通用人脸检测模型,可以在低算力设备中如用ARM进行实时的通用场景的人脸检测推理。
PaddleHub对于支持一键预测的module,可以调用module的相应预测API,完成预测功能。
(1)首先安装PaddleHub
!pip install paddlehub==1.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
(2)加载预训练模型
import paddlehub as hub
module = hub.Module(name="ultra_light_fast_generic_face_detector_1mb_640")
(3)拿到返回结果,设置人脸检测门限,统计图片中总人数。
input_dict = {"image": test_img_path}
# execute predict and print the result
results = module.face_detection(data=input_dict, visualization=True)
det = results[0]['data']
num = 0
for condition in det:
confidence = condition['confidence']
if confidence >= 0.9:
num = num +1
print("总人数:",num)
# for result in results:
# print(result)
# 预测结果展示
img = mpimg.imread("face_detector_640_predict_output/test_face_detection.jpg")
plt.figure(figsize=(10,10))
plt.imshow(img)
plt.axis('off')
plt.show()
三:预测结果
总人数:5
图片显示如下:
AI Studio 项目地址:https://aistudio.baidu.com/aistudio/projectdetail/449796
1
收藏
请登录后评论
一键点名【滑稽】
以后不能靠“替我点名的兄弟”了。。。
玩玩就好,千万别部署成服务,让千千万万的app能够轻易调用啊。。。
晚了,百度已经出了清点人数的API,一秒数几百人没压力哈
祈祷老师看不到
能轻点嗑书虫么和其他班“凑数”的么?哈哈
代码格式可以调一下吗
这样纯文本格式的看着很奇怪