基于YOLO V4-TLite的移动端君子兰病虫害检测方法

    Detection method for Clivia Miniata pests and diseases on mobile terminal based on YOLO V4-TLite

    • 摘要: 针对大棚和园林环境识别君子兰病虫害存在实时性差、检测精度低、过度依赖高算力和硬件功耗高等问题,提出一种面向移动端执行的YOLO V4-TLite君子兰病虫害检测方法。首先,以YOLO V4-Tiny为基础,使用低成本的部分卷积代替主干网络中的传统卷积。其次,使用逆残差网络结构,形成轻量化主干网络。再次,使用通道融合采样层机制,提升网络的鲁棒性和准确性。最后,将改进模型迁移部署在ROCK 5B移动端上,并针对君子兰3种典型病虫害叶枯病、黄斑病和介壳虫进行试验。试验结果表明,该改进模型的平均精度均值(mean average precision, mAP)为78.5%,内存占用量仅为4.8MB,浮点数运算量(floating point operations, FLOPs)为1.3 G,最大卷积计算的随机存储器(random access memory, RAM)储存为1 MB;桌面端单张检测速度为0.005 s,功耗为70 W;在移动端,CPU单张检测速度为0.239 s,功耗为10 W,NPU单张检测速度为0.018 s,功耗为7 W。YOLO V4-TLite模型在低资源和低功耗的移动端进行君子兰病虫害检测,其相比于现有主流YOLO系列模型具有较好的竞争力。

       

      Abstract: An accurate and rapid identification is required for the Clivia miniata pests and diseases in the greenhouse and garden in recent years. In this study, the YOLO V4-TLite algorithm was proposed to detect the Clivia miniata pest using a mobile terminal. The real-time performance and high accuracy were also achieved to reduce the over-reliance on the high-computing and high-power hardware. Firstly, the dataset of Clivia diseases and insect pests was collected from the real planting in the greenhouse. The images were captured at the early and middle stages of Clivia diseases and insect pests in winter and spring. Secondly, a low-cost improved partial convolution was used to replace the traditional one in the backbone network using the YOLO V4-Tiny model. The improved model was then obtained with the high speed of operation and the low consumption of memory. Thirdly, an improved structure of the inverse residual network was used to form a lightweight backbone network. The hardware compatibility was also enhanced to reduce the large consumption of random storage in the depth of the backbone network in the YOLO V4-Tiny model. The high operation speed of the model was obtained with the compatibility of the mobile terminal with the limited resources. Fourthly, the weight-sharing convolution was combined with the conventional convolution for channel fusion. The high robustness and accuracy of the network were obtained to reduce the redundant feature maps and their attention distraction in the traditional convolution layer of the YOLO V4-Tiny model. Finally, the improved model was deployed on the ROCK 5B mobile. Three types of Clivia miniata pests were then tested: leaf blight, maculopathy, and coccid. The experimental results showed that the better performance of the improved model was achieved with the mean average precision (mAP) of 78.5% at an intersection over union (IoU) ratio of 0.5, memory usage of only 4.8MB, and the floating point operations (FLOPs) of 1.3 G. The desktop single detection speed was 0.005 s with 70 W power consumption. On the mobile side, the CPU single detection speed was 0.239 s with 10 W power consumption. The NPU single detection speed was 0.018 s with 7 W power consumption. Compared with the original YOLO V4-Tiny model, the mAP50 of the YOLO V4-TLite model increased by 12.6 percentage point, whereas, the model size decreased by 78.6%. The computational efficiencies of the YOLO V4-TLite model were improved by 37.5 and 85.9 percentage point on the desktop and mobile side, respectively. While the power consumption demands were reduced by 26 and 2 W, respectively. The mAP50 values were 3.9, 2.3, 1.6, and 1.3 percentage point higher, respectively, compared with the target detection models of YOLOV11-N, YOLO V10-N, YOLO V7-Tiny, and YOLO V5-S. The YOLO V4-TLite model can be expected to detect the Clivia Miniata pest and disease on the low resource and power mobile. Better performance was also achieved, compared with the existing mainstream YOLO series models.

       

    /

    返回文章
    返回