首页|MMDGAN: A fusion data augmentation method for tomato-leaf disease identification
MMDGAN: A fusion data augmentation method for tomato-leaf disease identification
扫码查看
点击上方二维码区域，可以放大扫码查看
原文链接
NSTL
Elsevier
Tomato disease control is of great significance to ensure crop production and tomato disease classification study is an essential tool for doing so. In this paper, we propose a new data augmentation method based on deep threshold multi-feature extraction convolution GAN with Mixed Attention and Markovian Discriminator (MMDGAN) for tomato disease leaf classification. Firstly, in the generator of MMDGAN, a deep threshold multi-feature extraction module was proposed to improve the feature extraction of tomato disease leaves. Then, a mixed attention mechanism combined cross attention module with fused features-highlighting module was proposed to coordinate the overall generation of images. Finally, for the discriminator, Markov discriminator was used to strengthen the similarity judgment of local texture of images. Based on the open datasets PlantVillage, the Frechet Inception Distance (FID) score of healthy tomato leaf image, Leaf Mold, Leaf Curl and Spider Mite generated by MMDGAN were 159.3010, 164.4744, 230.3825 and 254.9866 respectively. Thereafter, a B-ARNet model is trained on synthetic and real images using transfer learning to classify the four categories of tomato diseases. The proposed method achieved an accuracy of 97.12%, with and F1 value of 97.78%. The proposed approach shows its superiority over the existing methodologies.
Deep threshold multi-feature extraction moduleMarkovian discriminatorMixed attentionMMDGAN data augmentationTomato leaf disease
Zhang L.、Zhou G.、Chen A.、Cai W.、Wang Y.、Li L.、Lu C.
展开 >
College of Computer & Information Engineering Central South University of Forestry and Technology Changsha
National University of Defence Technology Changsha
Department of Civil Architectural and Environmental Engineering Missouri University of Science & Technology
NSTL
Elsevier
