YOLOv4与树莓派平台的轻量化柑橘检测算法

• 作者: 韩进 舒媛
• 来源期刊: 单片机与嵌入式系统应用
• 关键词: GhostNet YOLOv4 轻量化 柑橘检测 树莓派
• 年，卷（期）: 2021,（7）
• 所属期刊栏目: 新器件新技术|NEW PRODUCT & TECH
• 页码范围: 46-50
• 页数: 5页
• 分类号: TP391
• 语种: 中文

摘要

为了满足高精度柑橘检测算法在嵌入式设备上的部署要求,论文基于YOLOv4与树莓派硬件平台设计了 一种高效的柑橘检测算法.首先借鉴GhostNet思想,设计了轻量级的网络单元并以此压缩了网络结构.其次,针对柑橘的形状特征用圆形边界框进行目标定位,同时改进了 CIOU边框回归损失函数.实验结果表明,改进后的算法在柑橘检测任务中的F1分数达83.95％,尤其对遮挡问题的改善效果明显,而模型大小仅为52.5 MB,移植到树莓派平台后对单张图片处理时间不足YOLOv4的三分之一,内存占用率仅为28.6％.