当前位置: 首页 > 工具软件 > Citrus > 使用案例 >

深度学习 Green Citrus Detection and Counting in Orchards Based on YOLOv5-CS and AI Edge System 论文篇

李和昶
2023-12-01

Green Citrus Detection and Counting in Orchards Based on YOLOv5-CS and AI Edge System


作者

华南农业大学电子工程学院

成果

实现了柑橘的检测与计数,算法的基础是YOLOv5和DeepSORT(完成在边缘设备上的部署)

创新之处

  1. 采用了数据增广,有水平垂直镜像、平移、模糊、旋转270度、添加噪声等。
  2. 在Focus层后的卷积层上添加CBAM注意力机制
  3. 在原来三个目标检测层的基础上,添加了一个小目标检测层(原来的特征图输出尺寸为52x52、26x26和13x13,现添加了104x104的尺寸)。
  4. 将GIoU损失函数改为CIoU损失函数
  5. 将算法部署在了边缘设备中。
  6. 学习率的调整采用了余弦退火算法(YOLOv5自带的)。
  7. 在跟踪计数中,引入了“虚拟线”或者“虚拟区域”的概念(跟实验室同学的思路类似),即柑橘穿过一条设定的线,柑橘数目就增加。
 类似资料: