[发明专利]一种基于改进YOLOV4的密集小目标检测方法

说明书

本发明公开了一种基于改进YOLOV4的密集小目标检测方法，其特征在于通过Mosaic数据增强方法对密集小目标检测数据集在数据广度层面进行数据特征融合处理，丰富检测物体的密集程度以及检测物体的背景信息和目标信息，最后对Mosaic数据增强方法处理后的数据集进行密集小目标检测数据处理，使密集小目标数据集分布满足标准正态分布；应用改进YOLOV4密集小目标检测方法优化图像数据特征提取策略，丰富目标实例特征的语义信息，之后对密集小目标检测数据集进行迭代训练，使用随机梯度优化器对检测模型参数进行优化，将所得检测模型与目前最优检测模型进行对比，选择检测精度更高的模型作为当前最优检测模型并且继续进行迭代训练，经过多轮迭代训练之后生成最优密集小目标检测模型。本发明利用改进YOLOV4密集小目标检测方法的数据预处理机制以及特征提取策略生成最优的密集小目标检测模型，丰富目标实例特征的语义信息，有效地提高密集小目标检测精度。

技术领域

本发明属于目标检测技术领域和人工智能领域，具体涉及一种基于改进YOLOV4的密集小目标检测方法。

背景技术

目标检测是计算机视觉领域的基本任务之一，不仅需要对目标对象进行分类，而且需要预测目标对象所在位置。随着深度学习技术的发展以及深度神经网络在目标检测领域的广泛应用，当前目标检测技术在计算机视觉领域已经取得了相当不错的效果。然而在实际应用场景中，目标大多小而密集且存在目标高度重叠问题，现有目标检测技术其检测精度并不能满足实际需求，密集小目标检测仍然是具有挑战性的任务。

随着人工智能的高速发展，密集小目标检测方法现已在日常生活安全、交通场景检测、遥感图像检测、医学图像检测以及自动驾驶等领域有了初步的应用。因此有必要对密集小目标检测方法进行优化处理，生成更好的特征提取策略，建立最优的密集小目标检测模型，有效地提高密集小目标检测精度，为各种应用领域提供更好地服务。

发明内容

为了解决传统密集小目标检测方法存在的检测精度低、易漏检误检问题，本发明提供了一种基于改进YOLOV4的密集小目标检测方法。通过应用改进YOLOV4密集小目标检测方法的数据预处理机制以及特征提取策略生成最优的密集小目标检测模型，丰富目标实例特征的语义信息，有效地提高密集小目标检测精度。

为实现上述目的，本发明技术方案主要包括如下步骤：

1.建立密集小目标数据预处理机制：

通过Mosaic数据增强方法对密集小目标检测数据集在数据广度层面进行数据特征融合处理，丰富检测物体的密集程度以及检测物体的背景信息和目标信息，最后对Mosaic数据增强方法处理后的数据集进行密集小目标检测数据处理，使密集小目标数据集分布满足标准正态分布。

2.采用改进YOLOV4密集小目标检测方法生成最优检测模型：

(1)改进YOLOV4密集小目标检测方法将每张图像数据均匀划分为S×S的网格，并且每张图像数据的每个网格由B个边界框来进行预测；设定改进YOLOV4密集小目标检测方法中每张图像数据中目标实例位置标签为目标实例分类标签为目标实例置信度标签为目标实例预测位置结果为{x_i，y_i，w_i，h_i}，目标实例预测分类结果为p_i(c)，目标实例预测置信度结果为C_i；

(2)加载图像分类网络模型进行微调，之后进行密集小目标检测迭代训练，设定最大迭代次数为D，在每轮迭代中对图像数据集进行特征提取进行目标检测，之后通过损失函数L_YOLO计算该轮迭代损失，最后使用随机梯度优化器对模型参数进行优化；

(3)将本次迭代所得检测模型与目前最优检测模型进行对比，选择检测精度更优的检测模型作为当前最优检测模型并且进行迭代训练；

(4)重复步骤(2)、(3)，直到满足最大迭代次数D，密集小目标检测模型训练结束，得到最优密集小目标检测模型。