写点什么

18M 超轻量图像识别系统,商品、车辆、人脸识别一网打尽!

作者:百度大脑
  • 2022 年 1 月 28 日
  • 本文字数:1730 字

    阅读完需:约 6 分钟

图像识别作为深度学习算法的主流实践应用方向,早已在生活的各个领域发挥作用,如安全检查和身份核验时的人脸识别、无人货架和智能零售柜中的商品识别,这些任务背后的关键技术都在于此。

图 1 PP-ShiTu 应用于商品识别效果示意(开发者应用展示)

然而实现理想的识别效果并不是一件简单的事:

  • 针对海量数据问题而言,如何实现一个通用的方法在不同的数据集中都有很好的表征能力?

  • 不同物体间的差别可能极其微小,即使是同种物体,也会受到外界干扰呈现不同形态,究竟如何进行有效区分?

  • 识别需求更新频繁,使用单一的分类或者检测模型只能不断增加训练集重训模型,怎样才能降低开发成本,快速跟上迭代步伐?

PP-ShiTu 核心技术解读

针对以上技术难点,PaddleClas 推出的通用图像识别系统 PP-ShiTu 均可以完美解决,它不仅有高水平的特征学习能力,对新出现的类别也可以在不用重新训练算法的情况下,直接通过配置检索库的方式实现识别。

⭐ 项目链接 ⭐

https://github.com/PaddlePaddle/PaddleClas

PaddleClas 所有源码及教程均已开源,欢迎大家使用,并 Star 鼓励~

这样的特殊能力,得益于 PP-ShiTu 引入目标检测、度量学习、图像检索等技术,并将每个模块性能最大化,才构建成了统一、通用的图像识别系统:

图 2 PP-ShiTu 架构示意图

  • 主体检测:不同于一般的目标检测,主体检测是将目标物体和背景区分开来,能够有效减少干扰特征。PP-ShiTu 中所采用的是目标检测模型 PP-PicoDet,不仅预测速度超快,精度上也达到了 SOTA,为后续的准确识别打下坚实基础;

图 3 PP-PicoDet 在 COCO2017-val 上的 mAP 对比

  • 特征提取:作为图像识别系统的重点工作,PP-ShiTu 是采用 CPU 级轻量化骨干网络 PP-LCNet,并结合度量学习 arcmargin 算法,对高相似物体的区分效果远超单一模型,不仅准确率超越大模型 ResNet50,预测速度还能快 3 倍!再加上经过 18w 类数据集打磨的预训练权重,不用再费心选择,一个模型就可以实现多场景覆盖。

图 4 训练数据及部分测试数据示例

  • 向量检索:不同于单一模型直接输出类别信息,对于未知类别只有模型重训一个选择,耗时又耗力。PP-ShiTu 引入向量检索,通过计算目标物体和检索库中图像的相似度来输出类别信息,一次训练长期使用,高效又便捷。

图 5 端侧新类别建立索引流程(开发者应用展示)

升级版 PP-ShiTu 欢迎使用

时隔数月,PaddleClas 新版本 v2.3.1 为大家带来了更加轻量化的图像识别系统 PP-ShiTu。核心功能点如下:

  • 升级版 PP-ShiTu 大小仅 18M,完美支持移动端需求

  • 支持基于 C++的服务化部署,部署效率大幅提升

  • 支持移动端 Paddle Lite 部署教程,手机上也能轻松实现图像识别

图 6 手机识别效果展示(开发者应用展示)

PP-ShiTu 的使用也是十分简单、方便,在完成环境配置后只需三步即可完成快速体验:

第一步:下载 Inference 模型

第二步:构建索引

第三步:完成单张/批量图像识别

详细文档,请参考:

https://github.com/PaddlePaddle/PaddleClas/blob/release/2.3/docs/zh_CN/quick_start/quick_start_recognition.md

更有完整流程演示,可参考 AI Studio 项目:https://aistudio.baidu.com/aistudio/projectdetail/3445604

共建开源生态

自 PP-ShiTu 发布以来,引起了众多开发者的广泛关注。华东理工大学的高材生颜鑫,也是飞桨领航团的团长,带领团队基于 PP-ShiTu 开发了一套智能购物平台系统:通过图像即可精准识别顾客购买的商品,并返回完整的购物清单及应付价格,为智能货柜提供了非常好的视觉化解决方案。

链接指路:

https://github.com/thomas-yanxin/Smart_container

图 7 智能结算平台架构示意图

与此同时,我们也力求为开源社区的繁荣作出贡献,希望 PP-ShiTu 的算法本身以及优化思路,可以在给业界开发者带来更好网络的同时,也带来更多的算法优化启发,为中国的开源发展贡献一份力量。

微信扫描二维码添加运营小姐姐,添加成功后,回复暗号【c】,即可收到邀请官方交流群邀请,在这里您可以获得更高效的问题答疑,与各行各业开发者充分交流,期待您的加入!

如果您想详细了解更多飞桨的相关内容,请参阅以下文档。

官网地址:

https://www.paddlepaddle.org.cn

PaddleClas 项目地址:

GitHub: 

https://github.com/PaddlePaddle/PaddleClas/tree/develop

Gitee: 

https://gitee.com/paddlepaddle/PaddleClas

这么强大、用心的项目 ,还不赶紧给各位开发者一个 Star🌟的鼓励!

用户头像

百度大脑

关注

用科技让复杂的世界更简单 2020.07.15 加入

百度大脑是百度技术多年积累和业务实践的集大成,包括视觉、语音、自然语言处理、知识图谱、深度学习等 AI 核心技术和 AI 开放平台。 即刻获取百度AI相关技术,可访问 ai.baidu.com了解更多!

评论

发布
暂无评论
18M 超轻量图像识别系统,商品、车辆、人脸识别一网打尽!