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

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. 在跟踪计数中，引入了“虚拟线”或者“虚拟区域”的概念（跟实验室同学的思路类似），即柑橘穿过一条设定的线，柑橘数目就增加。