基于卷积神经网络的轻量化目标识别算法研究

来源 :长春理工大学 | 被引量 : 0次 | 上传用户:nicolaskerry
下载到本地 , 更方便阅读
声明 : 本文档内容版权归属内容提供方 , 如果您对本文有版权争议 , 可与客服联系进行内容授权或下架
论文部分内容阅读
目标识别是计算机视觉中一项重要的技术,随着深度学习的兴起,基于卷积神经网络的目标识别算法弥补了传统算法在识别精度、速度方面的不足。然而,基于卷积神经网络的目标识别算法网络计算复杂、生成模型较大,占用了设备大量的算力资源和存储空间,在无GPU的嵌入式设备上应用,识别精度和速度未能达到理想的效果。基于上述问题,论文以主流网络YOLOv3为基础,从提高算法识别准确率和对识别算法模型轻量化两个方面进行研究。针对目标识别准确率方面,论文设计了基于YOLOv3改进的高精度目标识别算法CFPN_YOLO。CFPN_YOLO通过改进原有多尺度融合预测的方法,调整输入尺寸并在倒数第四层增加一个预测层进行更多层的特征融合,使网络获得图像更丰富的底层细节信息;引入一致性监督损失函数,减少特征求和前不同预测层之间的信息丢失,提高算法的识别率。将改进算法在NWPU VHR-10和Aug-Fruits360两个数据集上分别进行测试,并与其他算法进行对比。实验结果表明,改进后算法的识别准确率相比YOLOv3分别提高了12.9%和15.1%。针对目标识别算法轻量化方面,论文首先在改进CFPN_YOLO算法的基础上设计了轻量化识别算法LDS_YOLO。LDS_YOLO结合深度可分离卷积对网络卷积块进行改进,将传统卷积替换成3×3和1×1的点卷积,加深网络深度减少模型参数;引入密集连接模块替换残差模块,增加不同层之间的关联度,避免过拟合与梯度消失的问题。改进算法选择用Aug-Fruits360数据集完成识别模型训练,并在无GPU的树莓派3B+平台上进行测试,分析其在目标识别精度、速度和参数量等方面的提高和优化程度。实验结果表明,改进后算法的识别准确率相比YOLOv3提高了10.7%,模型参数量平均只有8M,计算量为125 MFLOPs。然后,通过改进特征通道权重的剪枝操作,进一步对模型进行轻量化,在保证识别率的基础上将模型参数量降低为3.6M,计算量为77MFLOPs,测试识别率比原始模型损失不到2.3%,速度达到17帧/秒。最后,通过在其他公共数据集上进行训练和测试,并与主流识别算法进行对比,分析轻量化后算法模型的抗过拟合能力和泛化能力,证明基于卷积神经网络的轻量化目标识别算法的有效性和可行性。
其他文献
光电探测器将光辐射转换成电信号,根据探测波段的不同,可以分为紫外探测器,红外探测器,和宽波段探测器等。紫外光电探测器因具有较高灵敏度和快速的时间响应,得到了广泛关注。宽禁带半导体材料辐射耐受性强、热导率大、击穿电场强度高、电子迁移率高、化学稳定性好,已经成为制备紫外探测器件的主导材料。二维p型铜铁矿材料禁带宽度大于可见光子能量上限3.11 e V,依赖空穴导电。其中的CuGaO2微米片禁带宽度3.
在“十四五”发展规划中,智能制造是工业发展的必然趋势,光学镜头是智能制造中不可或缺的一部分。镜片作为光学镜头的重要组成部分,镜片质量的好坏对镜头性能有着举足轻重的影响。在镜片生产过程中,难免会产生各种类型的疵病,传统人工目视检测方法效率低、成本高、稳定性差。依赖于人工提取疵病特征的机器视觉检测方法,通过机器代人降低了检测成本,提高了稳定性,但仍严重依赖于检测经验,且受环境影响较大。基于深度学习的疵