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

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

    • 摘要: 针对大棚和园林环境识别君子兰病虫害存在实时性差、检测精度低、过度依赖高算力和硬件功耗高等问题,提出一种面向移动端执行的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: Aiming at the problems of poor real-time performance, low detection accuracy, and over-reliance on high-computing and high-power hardware for identifying Clivia miniata pests and diseases in greenhouse and garden environments, Proposing a YOLO V4-TLite Clivia miniata pest detection method for mobile execution. Firstly, the dataset of Clivia diseases and insect pests comes from the real greenhouse planting environment. The picture collection time is winter and spring. The degree of disease includes the early and middle stages of Clivia diseases and insect pests. Secondly, based on the YOLO V4-Tiny model, a low-cost improved partial convolution is used to replace the traditional convolution in the backbone network, so that the improved model can have faster operation speed and memory consumption. Thirdly, aiming at the problem of hardware compatibility caused by the large random storage consumption in the backbone network of the YOLO V4-Tiny model, an improved partial convolution and inverse residual network structure is used to form a lightweight backbone network, so that the improved model does not have large random storage consumption in the depth of the backbone network, which improves the operation speed of the model and the compatibility of the mobile terminal with limited resources. fourthly, given the problem that the traditional convolution layer of the YOLO V4-Tiny model has more redundant feature maps and feature map attention distraction, the weight sharing convolution and conventional convolution are used for channel fusion operation, so as to improve the robustness and accuracy of the network. Finally, the improved model migration was deployed on the ROCK 5B mobile and tested against three typical Clivia miniata pests: leaf blight, maculopathy, and coccid. The experimental results showed that the mean average precision (mAP) of 78.5% at an intersection over union (IoU) ratio of 0.5 for this improved model, with a memory usage of only 4.8MB, and the floating point operations (FLOPs) is 1.3 G. The desktop single detection speed is 0.005 s with 70 W power consumption. On the mobile side, the CPU single detection speed is 0.239 s with 10 W power consumption, and the NPU single detection speed is 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 %, and the model size decreased by 78.6 %. Compared with the YOLO V4-Tiny model, the computational efficiency of the YOLO V4-TLite model is improved by 35.7 % and the power consumption demand is reduced by 26 W on the desktop side. The computational efficiency on the NPU side of the mobile side is also improved by 85.7 % and the power consumption demand is reduced by 2 W. Compared with other backbone networks MobileNetV2 and GhostNetV1, its mAP50 is 3.7 % and 5.5 % higher, respectively. Compared with the target detection models YOLOV11-N, YOLO V10-N, YOLO V7-Tiny and YOLO V5-S, the mAP50 is 3.9 %, 2.3 %, 1.6 % and 1.3 % higher, respectively. The YOLO V4-TLite model performs Clivia Miniata pest and disease detection in a low-resource and low-power mobile, and it is competitive with the existing mainstream YOLO series models.

       

    /

    返回文章
    返回