基于PaddlePaddle的杂草模型量化部署与结构优化研究
Quantitative deployment of weed models based on PaddlePaddle Research on Structural Optimization
陈启 1任迎霞 1邓向武 2张威1
作者信息
- 1. 湖北黄冈应急管理职业技术学院信息安全系,湖北黄冈 438000
- 2. 广东石油化工学院电子信息工程学院,广东茂名 525000
- 折叠
摘要
针对农业中广泛应用的无人机等终端除草设备的计算资源少、存储资源有限等问题,文章通过对模型分别采取量化部署和更改模型注意力机制模块的方法来得到轻量模型.试验结果表明,移动端模型的推理时间只有服务器端模型的1/10左右,部署在树莓派3B+上的杂草分类推理速度是6fps左右,而且量化后的杂草分类模型规模有40％以上的减少,证明了量化对模型规模减少的有效性.同时基于CBAM注意力机制的MobileNet_CBAM杂草分类模型相比基于SE-Net注意力机制的MobileNetV3_Large模型在准确率上损失了0.3％,但模型参数规模降低了24.7％,整体性能更加均衡.本研究可为杂草模型的小型化落地应用提供理论基础和技术支持.
Abstract
In response to the problems of low computing resources and limited storage resources in terminal weeding devices such as drones,which are widely used in agriculture,this article ob-tains a lightweight model by quantitatively deploying and modifying the attention mechanism module of the model.The experimental results show that the inference time of the mobile mod-el is only about 1/10 of that of the server model.The weed classification inference speed de-ployed on Raspberry Pi 3B+is about 6fps,and the quantified weed classification model has a re-duction of over 40％in size,proving the effectiveness of quantification in reducing the model size.MobileNet based on CBAM attention mechanism simultaneously-CBAM weed classifica-tion model compared to MobileNetV3 based on SE Net attention mechanism_The Large model lost 0.3％in accuracy,but the model parameter size decreased by 24.7％,resulting in a more balanced overall performance.This study can provide theoretical basis and technical support for the miniaturization and landing application of weed models.
关键词
杂草识别/量化部署/注意力机制/结构优化Key words
weed identification/Quantitative deployment/Attention mechanism/Structural op-timization引用本文复制引用
基金项目
广东石油化工学院人才引进及博士启动项目(2019rc044)
出版年
2024
NETL
NSTL
万方数据