Spatial convolutional self-attention-based transformer module for strawberry disease identification under complex background

Abstract

The occurrence of strawberry diseases has a huge impact on the yield and quality of strawberry fruits, resulting in huge economic losses. Real-time and effective identification and diagnosis of strawberry disease is an essential step for strawberry disease prevention. Machine learning-based methods are widely used in strawberry disease identification tasks, but these methods require expertise to design proper strawberry disease feature descriptors. Deep-learning methods have remarkably improved the capability of feature extraction. However, the strawberry disease with complex backgrounds brings great challenges for accurate feature extraction, which leads to poor recognition results of strawberry disease under complex backgrounds. In this paper, an improved transformer-based strawberry disease identification method is proposed to achieve precise and fast recognition of multiple classes of strawberry diseases. First, a multi-classes strawberry disease dataset has been constructed with 5369 images and 12 types of common strawberry disease. To increase the diversity of samples under complex backgrounds, various data augmentation strategies are introduced into the strawberry disease recognition method. Then, Multi-Head Self-Attention (MSA) is used to capture feature dependencies over long distances of strawberry disease images by leveraging the self-attention mechanism. To improve the recognition efficiency, the spatial convolutional self-attention-based transformer (SCSA-Transformer) is proposed to reduce the parameters of the transformer network. The experimental results validated on the constructed strawberry disease dataset demonstrate that the recognition accuracy of the proposed method can achieve 99.10%, which outperforms other methods. Besides, we also observe that the parameters of the classification model are reduced compared with other methods, which effectively improves the recognition efficiency of strawberry diseases.

Plant phenotyping relevance

イチゴ葉・植物体の病害状態を画像から識別する深層学習手法を開発し、専用データセットで性能検証しており、植物病害フェノタイピング手法が中心である。

Code and data availability

公開状態または取得可能な本文経路を確認できませんでした。

No evidence-backed public reproduction asset is currently recorded.