Cassava Leaf Disease Classification Based On CNN
DOI:
https://doi.org/10.54097/hset.v16i.2363Keywords:
Convolutional Neural Network, Resnet, Image Classification, Data Augmentation.Abstract
Cassava, a key food security crop in Africa, can provide a large amount of carbohydrates, which is widely grown in Africa. This kind of crop has been disturbed by viral diseases for a long time, which is the main reason for low production. The symptoms of cassava diseases can be reflected in their leaves but artificial visual diagnosis of diseases is inefficient. In order to get rid of this inefficient method, it is important to establish a model to automatically diagnose cassava by using the captured crop images to help farmers save their crops. Convolutional neural network has a good performance in image classification. Based on the research of Kaiming He et al. on residual convolution network, this paper trains the image dataset of cassava leaves from Kaggle by residual networks and then compares the classification effects by using different layers. Moreover, data augmentation and Focalloss function are used to optimize the model, which achieve better performance.
Downloads
References
Rosenblatt F. The perception: a probabilistic model for information storage and organization in the brain[J]. Psychological review, 1958, 656: 386-408.
Rumelhart D E, Hinton G E, Williams R J. Learning representations by back-propagating errors[J]. Nature, 1986, 323: 533-536
HINTON G E, SALAKHUTDINOV R. Reducing the Dimensionality of Data with Neural Networks[J]. Science, 2006, 313: 504-507.
LecunY, Bottou L. Gradient-based learning applied to document recognition[J]. Proceedings of the IEEE, 1998, 86(11):227-2324.
KrizhevskyA, Sutskever I, Hinton G. ImageNet Classification with Deep Convolutional Neural Networks[J]. Communications of the ACM, 2012, 60: 84-90.
SzegedyC, Liu W, Jia Y, et al. Going Deeper with Convolutions[J]. 2015 IEEE Conference on ComputerVision and Pattern Recognition (CVPR), 2015: 1-9.
Zhou B,Khosla A, Lapedriza A, et al. Learning Deep Features for Discriminative Localization[J]. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2016: 2921-2929.
He K, Zhang X, Ren S, et al. Deep Residual Learning for Image Recognition[C]. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2016: 770-778.
A GéRon. Hands-on machine learning with Scikit-Learn and TensorFlow: concepts, tools, and techniques to build intelligent systems [C].O'Reilly Media, 2017.
SimonyanK, Zisserman A. Very Deep Convolutional Networks for Large-Scale Image Recognition[J]. ArXiv, 2015, abs/1409.1556.
Lin T Y, Goyal P, Girshick R, et al. Focal Loss for Dense Object Detection[J]. IEEE Transactions on Pattern Analysis & Machine Intelligence, 2020, 42:318-327.
Tan M, Le Q V. EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks[J]. ArXiv, 2019, abs/1905.11946.
Huang G, Liu Z, Laurens V, et al. Densely Connected Convolutional Networks[C].2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2017, pp. 2261-2269.
Srinidhi V V, Sahay A, Deeba K. Plant Pathology Disease Detection in Apple Leaves Using Deep Convolutional Neural Networks: Apple Leaves Disease Detection using EfficientNet and DenseNet[C]. 2021 5th International Conference on Computing Methodologies and Communication (ICCMC). 2021,pp. 1119-1127.
Downloads
Published
Issue
Section
License
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
How to Cite
