Real-Time Detection of Cucumber Leaf Diseases Based on Convolution Neural Network
Yiming Lou, Zelin Hu, Miao Li, Hualong Li, Xuanjiang Yang, Xianwang Liu, Fei Liu
Abstract
In this paper, we proposed an object detection algorithm based on deep learning for NVIDIA Jetson Xavier NX to detect cucumber leaf diseases in real time. We collected cucumber leaves images from the experimental base of Anhui Academy of Agricultural Sciences, and labeled dataset by VGG Image Annotatort. Based on the one-stage detection network YOLO v5, firstly, the size of anchors were obtained by cluster analysis of the dataset, and a variety of data augmentation was carried out for to enrich the background information, The feature fusion network is improved, and different scale features are given different weights. Based on these optimizations, our model achieved 84.6 mAP on our dataset with 34.7 MB parameters, 4.9 ms inference time on NVIDIA Geforce GTX1080Ti, and 23 FPS with 512×640 resolution on NVIDIA Jetson Xavier NX, being faster than previous detectors, basically met the purpose of real-time.