基于深度学习的目标检测网络的压缩及移植

来源 :南京信息工程大学 | 被引量 : 1次 | 上传用户:tomjerry2005
下载到本地 , 更方便阅读
声明 : 本文档内容版权归属内容提供方 , 如果您对本文有版权争议 , 可与客服联系进行内容授权或下架
论文部分内容阅读
目标检测是计算机视觉的重要组成部分,主要是定位一张图片或者视频中特定物体出现的区域并判断图像中出现的目标类型。随着深度学习的发展,基于深度学习的目标检测在众多领域中取得了令人瞩目的进步。然而这些基于深度学习的目标检测算法需要巨大的内存空间和高性能的计算设备,无法直接移植于资源受限的设备。因此,本文在总结现有研究方法的基础上,提出了一种改进的目标检测网络并将其移植到树莓派上进行目标检测,论文的主要研究内容如下:(1)目标检测过程分为目标的分类和定位,图像质量也决定了目标分类的精度,但现实应用过程中图像受到各种外界因素的影响,所以在对图像特征进行学习之前需要进行预处理。通过图像预处理可以消除图像中的无用信息,保留有用的真实信息,从而提高特征提取的可靠性和待检测目标的分类准确率。本文研究了图像灰度化,对比了不同阈值及阈值函数选择下的图像去噪效果,选取了最优的阈值及阈值函数进行小波去噪,得到更高质量的图像以提升模型的分类准确率。(2)提出一种联合扩张卷积和深度可分离卷积的轻量型神经网络。为了弥补当前深度网络参数量过大无法应用于资源受限设备和压缩网络导致准确率下降的问题,提出的一种基于扩张卷积和深度可分离卷积的轻量型神经网络,该网络不但极大的减少网络卷积运算中的参数和计算量,降低网络的大小,而且在保持每次卷积参数量不变的同时,增大卷积过程中的感受野,使得卷积过程中提取到更多全局且语义层次更高的特征,提高网络的分类准确率;此外,通过设置宽度因子,输入图片分辨率和扩张率三个超参数在保证较高准确率的前提下进一步降低模型的参数和计算量。实验结果表明,本文提出的方法能有效压缩网络大小,且得到较好的分类准确率。(3)针对目标检测算法SSD(Single Shot Multi Box Detector)利用低层特征图检测小目标容易导致特征提取不充分的问题,提出了基于空间金字塔的多尺度特征融合算法对传统SSD算法进行改进。首先,通过对低层特征进行多尺度卷积以增强深层特征提取;然后,融合得到的多尺度特征,使得得到的特征图能表征更多的信息。实验表明,该方法有效的提升了网络的检测精度。此外,将提出的联合扩张卷积和深度可分离卷积的轻量型网络作为改进后SSD的骨干网络,构造了一个基于改进SSD算法的轻量型目标检测网络,实验表明,在高性能计算机上,该网络能有效的权衡检测速度和检测精度之间的关系,在达到较快检测速度的同时拥有较高的检测精度。(4)为了将本文提出的目标检测网络应用于嵌入式等资源受限设备上,且能较好的实现目标检测任务。本文基于树莓派设备,采用Tensor Flow和Tensor Flow Lite两种方法对提出的模型进行移植,在树莓派上重建训练好的模型;然后通过设置树莓派摄像头模块实时采集图像并将每一帧图像输入到模型中进行目标检测。实验结果表明,该网络可以有效的在树莓派上进行目标检测,且Tensor Flow Lite框架能有效的提升模型的检测速度。
其他文献
电商平台为用户提供多种反馈方式,如评论、标签和评分等。评论作为评分的补充手段,包含更细粒度的反馈信息。商品有大量的属性特征,用户在评论中具体阐述其对商品各个属性的
随着智能机器人在环境感知、虚拟现实、医学诊断、自动驾驶等应用领域的不断拓展,对智能机器人环境感知和适应能力提出了更高的要求,仅拥有单一或单类传感器的系统难以满足机
随着科学技术与网络技术的迅猛发展,人类已进入大数据和人工智能时代。一方面,数据规模、类型、价值与时效的急剧变化,另一方面,模型场景、环境、需求与计算的日益复杂,人类
基于功能化纳米粒子修饰电极的电化学传感器由于制备方法简单、操作性强、选择性好、灵敏度高、响应时间短、检测成本低等优势,已被广泛应用于工业生产、临床医学、药物分析
如何安全高效地识别个人的身份,是信息化时代必须要解决的一个重要问题。传统的身份识别方法,如身份证、用户名等,存在着容易遗失和泄露等无法克服的缺点,越来越难以满足社会
推进纺织业智能信息化建设不但是国家战略也是行业需求,其中织物疵点的智能自动检测是重中之重,占据重要地位。因其作为纺织业质量控制的重要环节,直接影响产品质量,关系到企
深度估计在很久之前就在计算机视觉中得到很大的关注,到目前为止仍旧是一个有待继续研究的课题,因为深度估计不同于其他的计算机视觉算法,比如目标检测,人脸识别等,只需处理
骨龄评估通过骨骼发育形态推断儿童生长的实际情况,具有很多临床应用,例如诊断内分泌疾病和预测青少年最终的成年身高等。骨龄评估方法根据发展的历程可以分为:基于人工估计
作为全球范围内影响较为严重的生态环境问题之一,土壤侵蚀不仅对人类的生存构成威胁,而且也阻碍了社会的发展。土壤侵蚀的防治,必须建立在土壤侵蚀定量评价制图、掌握土壤侵
随着机器人技术的快速发展以及机器学习等技术的逐渐成熟,移动跟踪机器人广泛应用于各个领域,在越来越多的岗位上代替人类完成了复杂工作,实现了生产生活的智能化。本文针对