中国农业机械化科学研究院集团有限公司 主管

北京卓众出版有限公司 主办

基于改进YOLOv5的番茄品质识别技术

Tomato quality recognition technology based on improved YOLOv5

  • 摘要: 为提高番茄品质识别的精度与速度,提出一种基于改进YOLOv5的识别方法。以YOLOv5为基础识别框架,通过Bottleneck Transformer(BoT3)模块替换YOLOv5中CSPDarknet53网络的空间卷积模块,从而提高网络计算速度;引入注意力机制增强YOLOv5的特征表达能力,并使用SIoU损失函数替换YOLOv5的CIoU损失函数;最后,利用改进YOLOv5对番茄品质进行识别。结果表明,所提方法对红色和绿色番茄品质的平均检测精确率、召回率、全类平均精度分别达到90.11%、95.21%和95.10%,平均检测速度129帧/s;相较于SSD、CNN和VGG,改进模型具有明显的识别优势。研究表明,该改进方法可提高番茄等农作物品质的识别精度和识别速度。

     

    Abstract: To improve accuracy and speed of tomato quality recognition, a recognition method based on an improved YOLOv5 was proposed.Method was based on YOLOv5 as a recognition framework.It replaced spatial convolution module of CSPDarknet53 network in YOLOv5 with Bottleneck Transformer(BoT3)module to improve network computing speed.Attention mechanism to enhance feature expression ability of YOLOv5 was introduced, and replacing CIoU loss function of YOLOv5 with SIoU.Finally, improved YOLOv5 was used to identify tomato quality.Results showed that proposed method achieved an average recognition accuracy, recall, and accuracy of 90.11%, 95.21% and 95.10% for red and green tomatoes' quality, with an average detection speed of 129 frames per second.Compared with SSD, CNN and VGG, improved YOLOv5 had obvious recognition advantages.It was proved that improved model could improve recognition accuracy and speed of crop quality of such as tomatoes.

     

/

返回文章
返回