基于改进MobileNetV2的番茄病害识别研究

摘要

[目的]本文提出一种具有新轻量化瓶颈层模块和纹理注意力的TB-MobileNetV2轻量级网络，旨在解决多类番茄不同病害程度识别研究较少、识别精度较低、收敛速度较慢等问题。[方法]参照设计高效网络的相关准则设计了更高效轻量的瓶颈层模块设计即陀螺块(TOP Block),使用对病害纹理特征具有更高效关注的改进纹理注意力模块(texture coordinates attention)。[结果]TB-MobileNetV2对包含16种类别的番茄早晚期病害数据集的测试集识别准确率为88.49%,较原MobileNetV2网络提升了2.18%,且具有计算量减少与单张图片运行时间基本相同的优势；同时其准确率和收敛速度也均优于ShuffleNetV2 1.0×、MobileNetV3-Small、MobileNetV2和MobileNeXt等轻量级网络。[结论]本文提出的TB-MobileNetV2能有效提高对多种番茄早晚期病害识别准确率。

关键词

图像处理 / 多类番茄病害 / 早晚期识别 / 陀螺块 / 纹理注意力