百度AI攻略:Paddlehub人脸检测

PaddleHub可以便捷地获取PaddlePaddle生态下的预训练模型,完成模型的管理和一键预测。配合使用Fine-tune API,可以基于大规模预训练模型快速完成迁移学习,让预训练模型能更好地服务于用户特定场景的应用。

本次介绍如何使用paddlehub实现人脸检测。

模型概述

Ultra-Light-Fast-Generic-Face-Detector-1MB是针对边缘计算设备或低算力设备(如用ARM推理)设计的实时超轻量级通用人脸检测模型,可以在低算力设备中如用ARM进行实时的通用场景的人脸检测推理。该PaddleHub Module的预训练数据集为WIDER FACE数据集,可支持预测,在预测时会将图片输入缩放为640 * 480。

代码如下:

import matplotlib.pyplot as plt

import matplotlib.image as mpimg

import paddlehub as hub

module = hub.Module(name="ultra_light_fast_generic_face_detector_1mb_640")

# 待预测图片

test_img_path = ["./crowd1.jpg"]

img = mpimg.imread(test_img_path[0])

# 展示待预测图片

plt.imshow(img)

plt.axis('off')

plt.show()

input_dict = {"image": test_img_path}

# execute predict and print the result

results = module.face_detection(data=input_dict)

for result in results:

    print(result)

# 预测结果展示

img = mpimg.imread("./face_detector_640_predict_output/crowd1.jpg")

plt.imshow(img)

plt.axis('off')

plt.show()

效果:

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 暖暖的阳光,蔚蓝的天空,金色的沙滩,翻滚的浪花……三年后当我再次踏入这个有风,有雨,有梦,却再不会有你的城...
    白雅星阅读 601评论 0 34
  • 翟湘云焦点网络第九期,2019年11月7日坚持原创分享613天。 今天轮到我值班,在这个校区内有三个年级,他们分别...
    香花盛开阅读 178评论 2 1
  • 今日的眼泪,没有能够止住,怀念我曾经的战友,一块奋战在工作岗位的伙伴,一个外表温柔,内里刚强,凡事忍让的好姐姐,因...
    穆易_生涯咨询师阅读 240评论 2 0
  • 引子:爱了你那么久,爱你对我产生的影响不是你能想象到的。因为爱你,我的一生发生了不可言喻的改变…… 与你相识于幼年...
    似水流年gx阅读 424评论 0 1
  • 五点擂床,六点起床,七点出门溜达,八点上床睡觉。
    宇宙小福阅读 306评论 0 0