A Lightweight Image Super-Resolution Network Based on ESRGAN for Rapid Tomato Leaf Disease Classification

Lei Zha,Yangjing Shi,Juan Wen
DOI: https://doi.org/10.1007/978-981-16-6963-7_9
2022-01-01
Abstract:The Crop disease identification is an important task in intelligent agriculture. Image resolutions have a large impact on the overall accuracy of classification performance. Some crop diseases are so similar that low-resolution images cannot capture their differences. To reduce the losses caused by crop diseases, it is vital to study crop image super-resolution reconstruction. Recently, with the rapid development of deep learning, various Single Image Super-Resolution (SISR) methods based on convolutional neural network (CNN) have achieved remarkable performance. However, the existing SR networks mainly have large parameter sizes, which require numerous training images and computing resources. In this paper, a lightweight image super-resolution model is constructed and applied to tomato leaf disease identification. We introduce the Shuffle Blocks with an attention mechanism to replace the Residual in Residual Dense Blocks (RRDBs) in the Enhanced Super-Resolution Generative Adversarial Networks (ESRGAN), which is an effective SR model and won first place in the PIRM2018-SR Challenge. By the special structure we designed, our model can significantly reduce the parameter size of ESRGAN while almost maintaining its performance. Besides, we employ the tomato leaf images from the Plant Village dataset to train and test our model. Finally, we use the VGG16 to classify tomato leaf diseases based on the reconstructed images. The experiment results show that our model can effectively reduce the parameter size, computational complexity, and image reconstruction time compared to other chosen SR networks. Furthermore, the accuracy of SR images generated by our model is closer to ESRGAN and higher than other state-of-the-art methods.
What problem does this paper attempt to address?