科技频道百度飞桨开源业内首个口罩人脸检测及分类模型

科技频道 2020-03-24200未知admin

  抗击,众志成城,人工智能技术正被应用到防控中来。

  2 月 13 日,科技频道百度宣布免费开源业内首个口罩人脸检测及分类模型。该模型可以有效检测在密集人流区域中携带和未携戴口罩的所有人脸,同时判断该者是否佩戴口罩。目前已通过飞桨 PaddleHub 开源出来,广大者用几行代码即可快速上手,免费调用。

  模型可视化效果:绿框为佩戴口罩标注,红框为未佩戴口罩标注

  随着本周各企业相继复工,节后经济开始逐渐恢复,人脸口罩检测方案成为返中众多社区、大型厂商、央企的重要需求。如判断工区员工是否佩戴口罩、人流密集的关口运输中心如何识别戴口罩的人脸并测温、佩戴口罩是否也能完成日常刷脸打卡等等……都是新冠下需要解决的真实痛点。

  防控刻不容缓,百度此前已有多项 AI 举措助力防控。此次宣布免费开源的自研口罩人脸检测及分类模型,是基于 2018 年百度收录于国际计算机视觉会议 ECCV 的论文 PyramidBox 研发,可以在人流密集的公共场景检测海量人脸的同时,将佩戴口罩和未佩戴口罩的人脸快速识别标注。基于此预训练模型,者仅需使用少量自有数据,便可快速完成自有场景的模型。

  百度研发工程师介绍,口罩人脸检测及分类模型,由两个功能单元组成,可以分别完成口罩人脸的检测和口罩人脸的分类。经测试,模型的人脸检测算法基于 ceboxes 的主干网络加入了超过 10 万张口罩人脸数据训练,可在准确率 98% 的情况下,召回率显著提升 30%。而人脸口罩判断模型可实现对人脸是否佩戴口罩的判定,口罩判别准确率达到 96.5%,满足常规口罩检测需求。者基于自有场景数据还可进行二次模型优化,可进一步提升模型准确率和召回率。

  如此高的准确率的背后是大量数据训练的结果,新模型采用了超过十万张图片的训练数据,确保样本量足够且有效。另一方面,人脸检测模型基于百度自研的冠军算法,整个研发过程都是基于百度开源的飞桨深度学习平台,能够进行高效、便捷的模型、训练、部署。

  在线演示效果:绿色边界框为戴口罩人脸、红色边界框为不戴口罩人脸。感兴趣的者可自己上传图片测试模型的效果。

  在线演示地址:

  对于实际场景中的光照、口罩遮挡、表情变化、尺度变化等问题,模型具有鲁棒性,并且能够在多种不同端、边、云设备上实时检测,在落地过程中做到真正实用。

  模型开源,方便应用

  为了最大程度方便者应用,百度深度学习平台飞桨通过简单易用的预训练模型管理工具 PaddleHub 将人脸口罩检测模型开源出来,只需基本的 Python 编程能力,即可快速上手调用,如果具有一定的端 APP 能力,也可以快速将模型部署到端上。

  预训练模型介绍:

  预训练模型示例代码:

  这一方案中,用于识别人脸的模型基于 2018 年百度在国际计算机视觉会议 ECCV 2018 的论文 PyramidBox 而研发,基于自研的飞桨开源深度学习平台进行训练,并通过 PaddleSlim 等模型小型化技术使得算法能够高效运行在一些算力有限的设备上。

  此外飞桨还将提供海量二次的工具组件,科技频道以及更多的人脸相关检测算法,以上所有技术及工具都是开源且免费的。

  部署简单,上手迅速

  只要代码,我们就可以在自己的计算机处理口罩人脸检测。实际在我们采用 CPU 的情况下,检测基本是实时的,推断速度非常快。以下为调用预训练模型的核心代码,其中我们在当前文件夹下放了一张测试图像:

  # 设置输入数据

  更重要的是,作为一项完善的开源工作,除了本地推断以外,其还需要考虑如何将模型部署到服务器或设备中。若能快速部署到各平台,那么才真正意味着它可以作为「战疫」的基础工具。

  目前,百度提供了两个预训练模型,科技频道即服务器端口罩人脸检测及分类模型「pyramidbox_lite_server_sk」、以及端口罩人脸检测及分类模型「pyramidbox_lite_mobile_sk」,这两者能满足各种下游任务。

  1. 一条命令部署到服务器

  借助 PaddleHub,服务器端的部署也非常简单,直接用一条命令行在服务器启动口罩人脸检测与分类模型就行了:

  是的,在服务器端这就完全没问题了。相比手动配置各种参数或者调用各种框架,PaddleHub 部署服务器非常方便。相信只要有一些 Python 基础,在本地预测、以及部署到服务器端都是没问题的,飞桨的 PaddleHub 已经帮我们做好了各种处理过程。

  2. 三步部署到端

  Paddle Lite 是飞桨的端侧推理引擎,专门面向端的模型推理部署。如果我们需要把口罩人脸检测及分类模型嵌入到手机等设备,那么 Paddle Lite 这样的端侧推理引擎能帮我们节省很多工作。

  在端部署口罩人脸检测及分类模型,也只需要三步:

  下载预测库,Paddle Lite 会提供编译好的预测库;

  通过预测 API 实现调用。

  者可以通过 PaddleHub 下载人脸口罩识别模型。在正常安装 PaddleHub 以后,可以通过 Python 执行以下代码下载并保存模型,以下载保存端人脸口罩识别模型为例:

  通过以上代码,可以获得人脸检测和口罩佩戴判断模型,分别存储在 test_program 目录下的 pyramidbox_lite 和 sk_detector 子文件夹之中。文件夹中的model是模型结构文件,param文件是权重文件。

  其中比较重要的是端 API 调用方法,具体实现请参考下文给出的 Paddle Lite 的示例地址。

  人脸识别和佩戴口罩判断在端部署的示例地址为:

  作为开源、功能完备的产业级深度学习平台,开源是飞桨践行并努力的方向。新型感染的消息牵动着每个人的心,百度也在持续运用着 AI、大数据等前沿技术,助力“大后方”的防控。

  从成立总规模 3 亿元及公共卫生安全攻坚专项基金,到提供人工智能技术支持配套亿级计算资源;从百度地图为科学管理提供参考,到 AI 多人体温快速检测解决方案落地清河火车站、免费病毒线性时间算法、免费社区管能外呼平台……

  本次开源的的口罩人脸检测及分类模型,是百度 AI 能力的又一次倾力输出,也希望全更多的者一道加入,探讨,打赢这场阻击战。

  点个在看少个 bug

原文标题:科技频道百度飞桨开源业内首个口罩人脸检测及分类模型 网址:http://www.dzine-studios.com/kejipindao/2020/0324/31530.html

Copyright © 2002-2020 志同道合新闻网 www.dzine-studios.com 版权所有  

联系QQ:1352848661