结合主动光源和改进YOLOv5s模型的夜间环境柑橘检测方法

作者：熊俊涛; 霍钊威; 黄启寅; 陈浩然; 杨振刚; 黄煜华; 苏颖苗

来源：华南农业大学学报, 2024, (01): 1-19.

• 夜间柑橘检测
• 主动光源
• 双向特征金字塔网络
• YOLOv5s
• HSV颜色空间

【目的】夜间环境下柑橘的精准识别是实现采摘机器人全天候智能化作业的重要前提。为了解决夜间环境下遮挡和较小柑橘难以准确识别的问题。本文提出了一种结合主动光源的夜间柑橘识别方法。【方法】首先，通过分析主动光源下颜色特征不同的夜间柑橘图像，选择最佳的光源色并进行图像采集。然后，提出一种夜间柑橘检测模型BI-YOLOv5s，该模型采用双向特征金字塔网络Bi-FPN进行多尺度交叉连接和加权特征融合，提高对遮挡和较小果实的识别能力；并引入CA注意力机制模块，进一步加强对目标位置信息的提取；同时采用融入Transformer结构的C3TR模块，减少计算量和更好地提取全局信息。【结果】本文提出的BI-YOLOv5s模型在测试集上的精准率、召回率、平均准确率分别为93.4%、92.2%和97.1%，相比YOLOv5s模型分别提升了3.2、1.5和2.3个百分点。并且，在所采用的光源色环境下，模型对夜间柑橘识别的正确率为95.3%，相比白光环境下提高了10.4个百分点。【结论】本文提出的方法对夜间环境下遮挡和小目标柑橘的识别具有较高的准确性，可为夜间果蔬智能化采摘的视觉精准识别提供技术支持。

• 单位
  华南农业大学