首页 PaddleHub 帖子详情
PaddleHub人脸检测实现一键查人数
收藏
快速回复
PaddleHub 问答预训练模型 1582 8
PaddleHub人脸检测实现一键查人数
收藏
快速回复
PaddleHub 问答预训练模型 1582 8

    想必各位同学都经历过老师上课点名的场景,想逃不能逃才最寂寞……

    如今,老师们也可是用起了人工智能,仅需拍一张照片,就能一键查出今天来了多少人,是不是很便捷?怎么实现呢?其实也很简单了,下面做简要介绍:

    一、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
收藏
回复
全部评论(8)
时间顺序
自尊心3
#2 回复于2020-05

一键点名【滑稽】

0
回复
AIStudio810258
#3 回复于2020-05

以后不能靠“替我点名的兄弟”了。。。

0
回复
AIStudio810258
#4 回复于2020-05

玩玩就好,千万别部署成服务,让千千万万的app能够轻易调用啊。。。

0
回复
w
wangwei8638
#5 回复于2020-05
玩玩就好,千万别部署成服务,让千千万万的app能够轻易调用啊。。。

晚了,百度已经出了清点人数的API,一秒数几百人没压力哈

0
回复
w
wangwei8638
#6 回复于2020-05
以后不能靠“替我点名的兄弟”了。。。

祈祷老师看不到

0
回复
AIStudio810258
#7 回复于2020-05
晚了,百度已经出了清点人数的API,一秒数几百人没压力哈

能轻点嗑书虫么和其他班“凑数”的么?哈哈

0
回复
水水水的老师
#8 回复于2020-05

代码格式可以调一下吗

0
回复
水水水的老师
#9 回复于2020-05

这样纯文本格式的看着很奇怪

1
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户