基于YOLOv5s的移动端轻量化芒果果面缺陷检测应用设计

作者：聂衍文; 徐建^*

来源：武汉轻工大学学报, 2022, 41(05): 89-103.

• 目标检测
• YOLOv5s
• 芒果
• 轻量化
• 移动端应用
• 果面检测

基于YOLOv5s给出了适合移动端应用的芒果果面缺陷检测算法，并实现了相关应用程序的设计。应用程序所采用的IM＿YOLOv5s算法，使用PPLCNet和GhostNet构建骨干特征提取网络和加强特征提取网络，同时使用改进的EIOU损失函数代替原CIOU损失函数更好地定位边界框，改进后的算法相比于YOLOv5s算法浮点运算次数减少了约77.22%,参数量减少了约72.65%,模型大小减小了约70.80%,在消融实验环境下batch size设置为16时的推理时间减少了约38.12%。在移动终端上的测试表明，IM＿YOLOv5s可以比原算法推理时间减少约52.7%。

• 单位
  武汉轻工大学; 电子工程学院