前沿科技探索家 · 2020年12月28日

如何基于飞桨打造智能眼镜视觉辅助系统

v2-8def7b77a438f17cc5d9931d68dfd479_1440w.jpg
目前中国视障人士数量多达700万,占全世界视障人士总数的18%。此外,全国还有约1200万弱视人群,两者总和约占全国总人口的1.5%。严重的视觉障碍给他们日常生活带来了巨大困难。

视障人群需要专人陪同,或借助辅助工具例如导盲杖、导盲犬等才能正常生活。这些传统解决方案,始终存在普及率低、成本高以及使用困难等问题。如何有效助力视障人士日常生活与出行,改善他们的生活质量,是当下全社会亟待解决的问题。

v2-f704de7824e7948b716f183f477e931d_1440w.jpg
图1 视障人士的困境(图片来自网络)
幸运的是,当前广泛使用的手机芯片,不仅具备高效能、超低功耗、开发设计弹性,也满足特殊行业应用的定制系统整合设计需求,因此非常适合用来开发智能出行类产品应用。例如,手机搭配百度端侧推理引擎 Paddle Lite,不仅能够加速设备开发,同时能提供设备持续维护、优化的设计弹性,加上目前手机完善的生态链和自身高可靠度表现,可大大降低智能眼镜设备开发过程的风险。

此外,通过设计出合理的并行运算机制,可以对人工智能算法进一步优化,特别是那些利用深度学习技术进行训练或推理的场合。百度飞桨支持并行计算,可以很容易设计出符合要求的系统结构;其丰富的程序资源和硬件支持,可以满足更广泛的市场需求。

因此,我们基于人工智能技术,开发出一套视觉辅助智能眼镜系统。该系统首先通过百度深度学习平台飞桨进行大规模日常生活环境图像的训练,形成服务于视障人士生活与出行的深度神经网络。接着配合智能眼镜,采用安卓智能手机作为运算核心,使用 Paddle Lite 作为智能手机上的人工智能图像推理平台,并导入前述网络配置参数,从而在智能眼镜系统中实现对视障人士的生活和出行提供辅助。

用户佩戴上眼镜后,语音告诉眼镜相应的指令,例如,进行物体寻找,通过交互模块,手机端首先会处理用户的语音信号,并执行相应的指令;与此同时,智能眼镜通过眼镜镜架处的摄像头捕捉周围的环境,通过通信模块传入手机端,手机采用飞桨生成的深度卷积神经网络 SSD 作为核心算法单元完成画面中的物体检测与识别。获得结果以后,将结果通过交互模块再反馈给用户。

如何实现智能眼镜系统

v2-a1a74394565fb8be70203a6e6ac73bae_1440w.jpg
图2 智能眼镜系统的基本流程
本系统基于深度学习技术中的深度卷积神经网络(Deep Convolution Neural Network),以目前通用的日常生活环境图像为训练数据集,开发出一套应用于视障人士的智能服务导航系统。本系统的软件核心算法通过飞桨进行卷积神经网络搭建,硬件部分采用安卓智能手机以及配有摄像头的眼镜,利用 Paddle Lite 进行安卓手机端的开发,完成了如下两方面的设计工作:

1、飞桨深度学习算法设计,即通过飞桨生成用于辅助视障人士日常生活和出行的深度神经网络。
2、安卓程序开发,即读取摄像头图像,使用 Paddle Lite 在智能手机上进行神经网络推理等功能开发,最后进行系统测试。
v2-a63efd8d1e5a8469ebf6f3a381052eb7_1440w.jpg
图3 利用飞桨和智能眼镜帮助视障人士进行智能导航
数据图像训练样本

如下图图4所示,系统中用于训练的样本选用平时常用的生活用品,如水杯,瓶子,自行车,汽车等20类图片,其中包含9963张标注过的图片,由训练集、测试集、验证集三部分组成,共标注出24640个物体。

对于非视觉障碍的人而言,从图中可以直观地分类出这些内容。图像经过初步的预处理,根据数据集中提供的标记信息,传入深度卷积神经网络进行训练。

v2-b3df002ce101b8d8b55f586e4514af70_1440w.jpg
图4 训练样本图片示例
系统实现

整个系统由三个交互的主题构成:眼镜、手机以及用户,而在具体实现过程中,我们将系统分为如下几个部分:

通信模块。负责连接眼镜上的摄像头进行传输,将视频数据传给手机端。
分析模块。是最为核心的部分,包含 SSD 深度神经网络,用于完成物体检测与识别。
交互模块。完成实时播报和语音交互。
v2-42f8d83f61f40cf27488013a24c0f5d4_1440w.jpg
图5 分析模块 SSD 深度神经网络结构
分析模块的 SSD 算法的特点在于 SingleShot,也就是说只要观察一次图片,就能够进行目标检测和识别,因此算法的效率非常高。SSD 网络属于典型的回归目标检测算法,具体分为四个步骤:

通过深度神经网络提取整个图片的特征。
对于不同尺度的深度特征图设计不同大小的特征抓取盒。
通过提取这些特征抓取盒对应的深度特征图的特征进行目标识别。
在这些识别的结果中通过非极大值抑制选择最佳的目标识别结果。
这里的多个特征抓取盒就是多尺度的,换言之,SSD 算法的核心思想就是从深度神经网络的不同层来提取特征,并分别利用这些特征进行回归预测。

本系统中,涉及到的 SSD 算法的代码部分如下所示:
v2-c64c6e13144b294e7357e3f039b6ad1e_1440w.jpg

为了提升模型性能和加快训练速度,我们在主模型中插入了 mobileNet 预训练模型,mobileNet 模型结构如下:
v2-480f3f0301c6a3aeb500ea1f2f3e5c30_1440w.jpg
在输出层,将 softmax 作为分类输出函数,损失函数使用交叉熵(cross_entropy)函数,采用在均方根传递算法(RMSprop)上改进的适合比较大规模的适合训练大数据集的阶梯阶梯型的学习率算法,通过反向传播来不断更新模型中的参数,从而使得损失函数逐渐减小来不断优化模型。

SSD 网络训练

  1. 数据准备

对于 SSD 网络的训练,我们选择从 ImageNet 官网下载 PASCAL-2012 的图像数据,共计17125张图片。依7:1:2的比例将数据集分为训练集 train_dates、验证集 validate_date 和测试集 test_date, 分别放入到"train" 、 "val" 和”test”目录中,供之后使用。

  1. 模型训练

首先从 paddle_model 中导入采用已经预训练好的 mobileNet 模型,用 train 中的60000张图片进行再训练,设置类别数量 class_num=80,batch_size=256, 在0.1的 learning rate 下训练100个 epoch。然后以7e-5的 decay_rate 衰减 learning_rate,微调模型参数,再训练100个 epoch,最终得到参数收敛的模型,表示训练已完成。
v2-fe7998d5ea272ec41405b7723640f02c_1440w.jpg

图6 SSD 网络训练损失与训练迭代次数之间的关系
可以看出,在训练过程中,尽管在短时间跨度内,训练误差有着剧烈的波动,但在多次的迭代下,总体呈现下降的趋势,并最终趋于稳定。

MobileNet 模型链接
https://github.com/PaddlePadd...
飞桨模型库链接:
https://github.com/PaddlePadd...

  1. 模型评估

· 系统性能评估

我们将设计好的眼镜系统进行性能评估,用于进行测试的物体识别的数据集由 VOC07 与我们自己拍摄的目标图像组成。目标检测的场景为视障人士日常生活中会遇到的相关场景进行识别,在本系统前期进行深度学习测试的过程中,采用飞桨所搭建的几种 SSD 卷积神经网络分别具有以下几种测试结果:

v2-e88ae69cc0480738e0bc4a4ea58bb8d4_1440w.jpg.png
表1 模型性能比较
不同网络的资源消耗不同,取得的识别结果也不尽相同。考虑到智能手机的资源情况,我们选择 MobileNet-SSD 作为核心,其在性能和资源消耗上可以取得很好的平衡。

· Paddle Lite 手机端部署

1)模型转化

首先,我们需要将训练后的神经网络模型的格式进行转化,使其适用于安卓端。对此,我们采用了百度新推出的为移动端贴身打造的 Paddle Lite,不仅能满足以上需求,还能对模型进行压缩,使其更小更快。

我们采用百度提供的 opt 对 PC 端的飞桨模型转化为移动端的 Paddle Lite 模型,在命令行中输入以下代码运行 opt 进行模型转换:
v2-6feba6712f12e78804f90e2a14ee7066_1440w.jpg

2)在安卓手机中使用 Paddle Lite 执行预测

主要流程有:

导入 baidu.paddle.liteJava 库。
利用 config.setModelFromFile 从路径中读取移动端的神经网络模型,然后将 config 设置为预测模型 predictor 的参数。
把一维 float 数组格式的输入 input 送入预测模型 predictor 中。
执行 predictor.run 运行神经网络进行预测。
最后,将 predictor 的输出 output 以一维 float 数组格式取出。
具体代码如下:
v2-c64c6e13144b294e7357e3f039b6ad1e_1440w.jpg

接下来,进行了实际场景测试。人的测试场景效果如图7和图8所示。
v2-ca477f7beef464eed0561d10b3eaf28c_1440w.jpg

图7 测试人物识别(近处场景)
v2-d2c534630d33a9d93a5ae042e48ff833_1440w.jpg
图8 测试人物识别(远处场景)
v2-975363caa0c905dc108bdf734b7ee4f3_1440w.jpg
图9 眼镜对于物体的检测识别结果
图中的方框标识出的物体是通过非极大值抑制进行输出的结果,也即取出 SSD 网络判断类型置信度最大的方框作为判断结果,实验结果表明,我们所设计的视觉辅助智能眼镜系统可以正确识别出所要检测的物体和目标。

基于飞桨打造的视觉辅助智能眼镜系统不仅能够帮助那些视障人士识别日常生活中使用的物体,同时还能快速获得出行指示并有效提示行人。这对于提高视障人士的生活质量、降低视障人士生活成本、缓解视障人士出行压力等方面着重要意义。同时,该系统也解决了“全国数量庞大的视障人士同数量有限、成本昂贵的辅助出行方式”之间的矛盾。

同时,本系统基于目前人工智能领域比较流行的深度学习技术进行物体图像识别,作品采用飞桨深度学习平台进行深度卷积神经网络的搭建,飞桨支持算法的高效设计以及深度神经网络的快速搭建,因此可以有效降低那些以往只能依赖人为判断的工作的复杂度,系统通过智能“学习”不同指示标识的特征,从而使得系统对新的出行指示标识能够进行正确的识别、判断。因此,本作品所设计的系统不仅可以用于日常生活物品识别,对于其他一些指示标志识别也具有一定借鉴意义。

更重要的是,本作品所设计的系统硬件部分基于安卓智能手机配合智能眼镜,这套硬件组合体积小、功耗低,对于视障人士十分便携。考虑到智能手机的普及程度,本作品所开发的智能眼镜系统便于推广,价格低廉,能够广泛深入到我国的各个地区,特别是偏远、经济不发达的地区,为提高未来视障人士生活质量,出行导航系统的小型化、便携化奠定了良好的基础,此外功耗低也可以满足设备续航能力以及环保节能的需求。

更多的实际演示效果可以登录链接参看本系统测试版本的视频:
https://www.bilibili.com/vide...

注:本文示例代码和视频所使用的是 Paddle Lite 早期版本,目前最新版本为 Paddle Lite v2.3,链接:

https://github.com/PaddlePadd...

如果想与更多深度学习同学交流,欢迎加入飞桨官方 QQ 群:703252161

如果在端侧部署中遇到问题,欢迎加入 Paddle Lite 官方 QQ 群与开发人员进行技术交流及问题反馈:696965088

推荐阅读
关注数
12978
内容数
325
带你捕获最前沿的科技信息,了解最新鲜的科技资讯
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息