基于深度学习的复杂背景下茶叶嫩芽检测算法

Detection algorithm of tea tender buds under complex background based on deep learning

针对传统的基于机器视觉的茶叶嫩芽检测方法存在手工特征提取鲁棒性较差以及准确率较低等问题,首次将基于深度学习的目标检测算法YOLO应用到复杂背景下的茶叶嫩芽图像的检测,并从多尺度检测方面对YOLO网络架构进行了改进,用大尺度和中尺度检测代替了原来的多尺度检测.在预处理阶段,通过结合超绿特征以及OSTU算法对复杂背景下的茶叶嫩芽图像进行了图像分割,使得茶叶嫩芽区域更加明显.实验结果表明,通过与其他算法对比,基于深度学习的目标检测算法对复杂背景下的茶叶嫩芽具有较高的检测精度,为复杂背景下茶叶嫩芽的智能化采摘设备的研究提供了基础.

The traditional tea bud detection method based on machine vision has the problems of poor robustness and low accuracy in manual feature extraction.The target detection algorithm YOLO based on deep learning is applied to the detection of tea tender image under complex background for the first time,and the architecture of YOLO network is improved from the aspect of multi-scale detection,using largescale and mesoscale detection instead of the original multi-scale detection.In the preprocessing stage,the image of tea shoots under complex background is segmented by combining the super green feature and the OSTU algorithm,which makes the tea tender bud area more obvious.The experimental results show that,by comparing with other algorithms,the target detection algorithm based on deep learning has a high detection precision for the tea tender shoots under complex background,which provides a basis for the research of intelligent picking equipment for tea tender shoots under complex background.