Green Citrus Detection and Counting in Orchards Based on YOLOv5-CS and AI Edge System
作者
华南农业大学电子工程学院
成果
实现了柑橘的检测与计数,算法的基础是YOLOv5和DeepSORT(完成在边缘设备上的部署)
创新之处
- 采用了数据增广,有水平垂直镜像、平移、模糊、旋转270度、添加噪声等。
- 在Focus层后的卷积层上添加CBAM注意力机制。
- 在原来三个目标检测层的基础上,添加了一个小目标检测层(原来的特征图输出尺寸为52x52、26x26和13x13,现添加了104x104的尺寸)。
- 将GIoU损失函数改为CIoU损失函数。
- 将算法部署在了边缘设备中。
- 学习率的调整采用了余弦退火算法(YOLOv5自带的)。
- 在跟踪计数中,引入了“虚拟线”或者“虚拟区域”的概念(跟实验室同学的思路类似),即柑橘穿过一条设定的线,柑橘数目就增加。