基于改进YOLOv5网络的串番茄成熟度的检测方法

本发明属于计算机视觉与模式识别，尤其涉及一种串番茄成熟度检测，具体为基于改进yolov5网络的串番茄成熟度的检测方法。

背景技术：

1、番茄营养价值丰富，有碳水化合物、维生素c、胡萝卜素、蛋白质以及多种营养素。串番茄，又名穗番茄，是近年来流行于国内外市场的新型番茄品种，具有着不错的市场。普通软果肉番茄的货架寿命最长为10天左右，而串番茄果实转红后能在果穗上保留15天以上而不软化、不裂果，货架寿命20天以上，有更长的保质日期。串番茄生长周期约三四个月，一年四季可实现滚动上市。因此，串番茄的采摘成为进入市场极为关键的一个环节。

2、对于温室大棚内种植的串番茄，目前采用的方式是由种植人员每日巡检，记录果实的生长状态，包括果实的挂果情况、果实的大小、果实的数量等。这类工作耗时耗力，占用大量人工成本。在智能化的工农业时代，智能化控制成为减少成本，提升生产效率的重要手段。

3、近几年，机器视觉领域的快速发展，也推动了农业领域向智能化生产的发展。但针对于大棚内串番茄的智能采摘，智能采摘设备的图像处理算法将识别出图像中的所有串番茄，而智能采摘设备一次只能采摘一串番茄，复杂场景下实现单个串番茄高效采摘是需要解决的问题。与此同时，对于串番茄的整串采摘，也需要考虑串番茄本身的质量水平，只有达到规定界限的串番茄才能整串采摘，减少生产损失，而对符合标准的串番茄的确定和识别，也成为了实际采摘过程中需要解决的问题。

技术实现思路

1、为了解决上述技术问题，本发明提供一种基于改进yolov5网络的串番茄成熟度的检测方法，能够实现对串番茄的精确识别，并进行进一步的成熟度检测，确定可采摘目标。相较于传统算法的目标检测，本发明为串番茄采摘的工程实际提供了具体的解决方案，具备逻辑性、完整性以及较高的精确性。

2、本发明所采用的技术方案是：

3、一种基于改进yolov5网络的串番茄成熟度的检测方法，包括以下步骤：

4、步骤1：构建串番茄数据集和单个小番茄数据集，所述单个小番茄数据集由串番茄中各个生长时期的单个小番茄的图像组成；

5、步骤2：基于改进yolov5网络分别对串番茄数据集和单个小番茄数据集进行训练，得到相应的权重模型配置，改进yolov5网络是将yolov5网络模型的backbone层的卷积层替换为ghostconv，head层输出头的detect替换为asff自适应特征融合层；

6、步骤3：基于改进yolov5网络，在detect程序中引入目标位置截取存储机制，构建串番茄检测模型detecttomato；

7、步骤4：串番茄检测模型detecttomato调用串番茄数据集的权重模型配置，识别出待检测图像中的串番茄检测目标，获得识别出的所有串番茄检测目标的图像及其位置信息，并将串番茄检测目标图像及其位置信息分别存储至exp文件夹和label文件夹中；

8、步骤5：基于改进yolov5网络，在detect程序中引入统计函数以及目标位置输出机制，构建成熟度检测模型detectmaturity；

9、步骤6：成熟度检测模型detectmaturity调用单个小番茄数据集的权重模型配置，遍历存储在exp文件夹中的所有串番茄检测目标图像，识别出每一幅串番茄检测目标图像中的单个小番茄，调用统计函数统计出同一幅串番茄检测目标图像中分别处于完熟期、半熟期、绿熟期的小番茄个数，将完熟期小番茄个数除以小番茄总数，计算得出串番茄成熟度，输出成熟度最高且达到预设成熟度标准值的串番茄检测目标图像对应label文件夹中的位置，从而确定采摘目标，在每完成一幅串番茄检测目标图像的成熟度检测后，都调用os库对label文件夹和exp文件夹进行清空处理。

10、本发明的有益效果包括：

11、1、明确单个采摘目标：本发明通过串番茄检测模型detecttomato，识别出图像数据中出现的串番茄，并将识别出的单个的串番茄检测目标图像以及位置信息进行存储，再通过成熟度检测模型detectmaturity对单个小番茄图像进行识别以及成熟度检测，明确最终采摘目标，为减少生产采摘损失以及采摘的流畅性提供了有效的方法；

12、2、小番茄果实成熟度分类：小番茄果实生长期共分为三个时期，包括绿熟期、半熟期和完熟期，三个时期中差异最大的是rgb分量中的r和g分量，即红色和绿色，本发明通过绘制三个生长时期的r-g分量统计表，成熟度检测模型detectmaturity基于三个生长时期的r-g分量统计表进行生长时期划分，为小番茄果实的成熟度检测提供了科学的依据，提高了串番茄成熟度检测和采摘的准确度。

13、3、轻量化网络模型：为了提升训练速度，将yolov5网络模型的backbone层的卷积层替换为ghostconv，head层输出头的detect替换为asff自适应特征融合层，在保证识别精度几乎不变的同时，训练时间减少了13％，一定程度上提升了应用价值。

技术特征：

1.一种基于改进yolov5网络的串番茄成熟度的检测方法，其特征在于，包括以下步骤：

2.根据权利要求1所述的一种基于改进yolov5网络的串番茄成熟度的检测方法，其特征在于，步骤2具体包括以下步骤：

3.根据权利要求1所述的一种基于改进yolov5网络的串番茄成熟度的检测方法，其特征在于，步骤4具体包括以下步骤：

4.根据权利要求1所述的一种基于改进yolov5网络的串番茄成熟度的检测方法，其特征在于，通过以下步骤构建串番茄数据集：

5.根据权利要求1所述的一种基于改进yolov5网络的串番茄成熟度的检测方法，其特征在于，通过以下步骤构建单个小番茄数据集：

技术总结
本发明涉及一种基于改进YOLOv5网络的串番茄成熟度的检测方法，包括步骤：构建串番茄数据集和单个小番茄数据集；基于改进YOLOv5网络对数据集进行训练，得到权重模型配置；构建串番茄检测模型DetectTomato，其识别出待检测图像中的串番茄检测目标，将串番茄检测目标图像及其位置信息分别存储在文件夹中；构建成熟度检测模型DetectMaturity，计算得出串番茄成熟度，输出成熟度最高且达到预设成熟度标准值的串番茄检测目标图像对应的位置，确定采摘目标，在每次成熟度检测后都清空文件夹。本发明能够实现对串番茄的精确识别，并进行进一步的成熟度检测，确定可采摘目标，提高了串番茄成熟度检测和采摘准确度。

技术研发人员：刘鹏,袁小龙,张建海,赵安然,李东齐,胡龙皓,宋楚轩
受保护的技术使用者：吉林大学
技术研发日：
技术公布日：2024/1/11