山东农业大学学报（自然科学版）2024，Vol.55Issue(3)：453-465,13.DOI:10.3969/j.issn.1000-2324.2024.03.018
面向无人机边缘计算的小麦麦穗计数轻量化模型研究
Lightweight Model of Wheat Ear Counting for UAV Edge Calculation
摘要
Abstract
The efficient counting of wheat ears is important for the rapid and accurate control of wheat yield.UAV is widely used in collecting field wheat information due to its high efficiency and low cost.However,the existing deep learning models for wheat ear counting have complex structure and large number of parameters,and cannot be directly deployed on the edge devices of UAVs with limited storage space.To address this issue,we proposed a model compression method combining pruning strategy and knowledge distillation,constructed a lightweight model based on YOLOv5s model,and designed a wheat ear count lightweight scheme for UAV edge computing.Experimental results show that the pruned and distilled YOLOv5s model achieves a counting accuracy of 93.3%and mAP of 94.4%in wheat counting tasks.The model size is reduced by approximately 76%,and the parameter count is decreased by 79.61%.Consequently,the model will take up less computing resources and storage space with high counting accuracy.The remarkable compression effect makes the model enables the model to be deployed on the edge equipment of UAV,providing the possibility for real-time counting of wheat ears.关键词
麦穗计数/无人机/深度学习/边缘设备/剪枝/蒸馏Key words
Wheat ear counting/unmanned aerial vehicle/deep learning/edge devices/pruning/distillation分类
信息技术与安全科学引用本文复制引用
刘旭,宋作杰,耿霞..面向无人机边缘计算的小麦麦穗计数轻量化模型研究[J].山东农业大学学报（自然科学版）,2024,55(3):453-465,13.基金项目
山东省自然科学基金面上项目(ZR2021MC168) （ZR2021MC168）
