基于批归一化与注意力机制的图像纹理识别算法  

Image Texture Recognition Algorithm Based on Batch Normalization and Attention Mechanism

作　　者：贺泽华 乔延松 赵绪营 赵耿[1] HE Zehua;QIAO Yansong;ZHAO Xuying;ZHAO Geng(Beijing Electronics Science and Technology Institute,Beijing 100071)

机构地区：[1]北京电子科技学院,北京100071

出　　处：《计算机与数字工程》2024年第3期646-652,共7页Computer & Digital Engineering

基　　金：北京电子科技学院“一流学科”建设项目(编号:1101014);国家自然科学基金项目(编号:61772047)资助。

摘　　要：针对传统图像纹理识别方法特征提取繁琐和纹理类间模糊性高、类内区分度低的问题,提出基于批归一化与注意力机制的卷积网络图像纹理识别算法。通过逐层批归一化将分散的数据统一,优化算法损失震荡和梯度消失问题;通过通道域和空间域的注意力机制对图像的关键区域和纹理的关键特征进行突出化表达。实验结果表明,所提算法不仅参数量低、计算速度快,而且在纹理数据集上的识别率达99.84%,超越基准模型和其他网络模型,证明该算法的对图像纹理具有良好的识别效果。Aiming at the cumbersome feature extraction of traditional image texture recognition methods,poor results,high in-ter-class ambiguity of texture,and low intra-class discrimination,a convolutional network image texture recognition algorithm based on batch normalization and attention mechanism is proposed.Through layer-by-layer batch normalization,the scattered data is unified,and the loss oscillation and gradient disappearing problems of the optimization algorithm are optimized.The key areas of the image and the key features of the texture are highlighted through the attention mechanism of the channel domain and the space domain.The experimental results show that the proposed algorithm model has low parameters and fast calculation speed.The recogni-tion rate on the dataset is 99.84%,surpassing the benchmark model and other network models,it proves that the algorithm has good recognition effect on image texture.

关 键 词：图像纹理 特征提取 卷积神经网络 批归一化 注意力机制 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]