基于改进YOLOv7的自然环境下柑橘果实识别研究

RESEARCH ON CITRUS FRUIT RECOGNITION BASED ON IMPROVED YOLOv7 IN NATURAL ENVIRONMENT

  • 摘要: 为实现柑橘果实的精准快速识别,提出一种改进的YOLOv7网络模型。结合ELAN结构与CA注意力机制,设计ELAN-CA结构,研究ELAN-CA结构与YOLOv7Backbone中的最佳结合点,并将对应位置上的ELAN结构进行替换,增强网络的学习能力,为了进一步提高网络的识别率,将CIOU损失函数替换为Focal-EIOU损失函数,解决损失函数的退化问题,模型平均识别精度从96.91%提升至97.56%。实验结果表明,CA注意力机制和Focal-EIOU函数能够有效提高YOLOv7在自然环境下识别柑橘果实的能力。

     

    Abstract: In order to achieve accurate and fast identification of citrus fruits, an improved YOLOv7 network model is proposed. The ELAN-CA structure was designed by combining the ELAN structure with the CA attention mechanism. The optimal combination point between the ELAN-CA structure and the YOLOv7 Backbone was studied, and the ELAN structure at the corresponding position was replaced to enhance the learning ability of the network. In order to further improve the recognition rate of the network, the CIOU loss function was replaced by the Focal-EIOU loss function, which solved the problem of loss function degradation. The model's average recognition accuracy was improved from 96.91% to 97.56%. Experimental results show that the CA attention mechanism and Focal-EIOU loss function can effectively improve the ability of YOLOv7 to identify citrus fruits in natural environments.

     

/

返回文章
返回