基于卷积神经网络和混合注意力机制的书标检测算法

张岩; 赵蒙蒙; 孙英伟; 常艳康

doi:10.16452/j.cnki.sdkjzk.2023.03.011

山东科技大学学报（自然科学版） ›› 2023, Vol. 42 ›› Issue (03) : 94 -102. DOI: 10.16452/j.cnki.sdkjzk.2023.03.011

基于卷积神经网络和混合注意力机制的书标检测算法

张岩, 赵蒙蒙, 孙英伟, 常艳康

作者信息 +

Author information +

文章历史 +

PDF (2089K)

摘要

为实现图书馆中机器人智能排架，提出一种基于卷积神经网络和混合注意力机制的书标检测模型。将DenseNet121引入YOLOv4以提高特征和梯度之间的传递效率，利用SPDC模块实现局部和全局特征融合，进而通过通道和空间混合注意力提高模型的特征表征能力。实验结果表明，模型的平均准确率、整体性能、参数量和模型大小均优于对比方法，且易于部署到嵌入式设备中实现在线检测，从而提高图书乱架治理的智能化水平。