吉林大学学报(理学版) ›› 2024, Vol. 62 ›› Issue (2): 381-0390.
吴叶辉, 李汝嘉, 季荣彪, 李亚东, 孙晓海, 陈娇娇, 杨建平
WU Yehui, LI Rujia, JI Rongbiao, LI Yadong, SUN Xiaohai, CHEN Jiaojiao, YANG Jianping
摘要: 针对图像识别中获取全局特征的局限性及难以提升识别准确性的问题, 提出一种基于随机增强Swin-Tiny Transformer轻量级模型的图像识别方法. 该方法在预处理阶段结合基于随机数据增强(random data augmentation based enhancement, RDABE)算法对图像特征进行增强, 并采用Transformer的自注意力机制, 以获得更全面的高层视觉语义信息. 通过在玉米病害数据集上优化Swin-Tiny Transformer模型并进行参数微调, 在农业领域的玉米病害上验证了该算法的适用性, 实现了更精确的病害检测. 实验结果表明, 基于随机增强的轻量级Swin-Tiny+RDABE模型对玉米病害图像识别准确率达93.586 7%. 在参数权重一致, 与性能优秀的轻量级ransformer、 卷积神经网络(CNN)系列模型对比的实验结果表明, 改进的模型准确率比Swin-Tiny Transformer,Deit3_Small,Vit_Small,Mobilenet_V3
Small,ShufflenetV2和Efficientnet_B1_Pruned模型提高了1.187 7%~4.988 1%, 且能迅速收敛.
中图分类号: