融合通道和空间注意力的茶叶病害目标检测方法

Tea disease detection method using fusion of channel and spatial attentions

  • 摘要: 准确、快速检测茶叶病害对茶叶产业经济意义重大。然而,病害叶片尺度变化、相互遮挡、复杂背景等问题大大降低了检测精度。提出了一种融合通道和空间注意力的茶叶病害目标检测方法,以实现准确、高效的病害检测。首先,利用多层的自注意力机制提取茶叶病害的多尺度特征,以获取茶叶图像在不同尺度下的局部细节特征;其次,引入新的通道注意力模块以提取更丰富的通道信息,同时抑制复杂背景噪声并增强模型特征表示能力。此外,设计一种新的空间注意力模块进一步提取特征的空间关系、减少冗余信息并优化计算开销。实验结果表明,融合通道和空间注意力的茶叶病害检测方法能应对叶片的尺度变化、相互遮挡和背景复杂等挑战。

     

    Abstract: Accurate and rapid detection of tea disease is of great economic significance to the tea industry. However, the problems such as scale variation, mutual occlusion and complex background of disease-affected leaves greatly reduce the prediction accuracy of the detector. A tea disease target detection method, integrating channel-spatial attention, was proposed to achieve accurate and efficient disease detection. Firstly, the multi-layer self-attention mechanism was used to extract multi-scale features of tea diseases to obtain local detail features of tea images at different scales. Secondly, the new channel attention module was introduced to extract richer channel information, suppress complex background noise and enhance the ability of model feature representation. In addition, a new spatial attention module was also proposed to further extract the spatial relationship of features, reduce redundant information and optimize computational overhead. The experimental results showed that the tea pest detection method with channel-spatial attention could cope with the challenges of leaf scale changes, mutual occlusion and complex background.

     

/

返回文章
返回