[发明专利]一种数据平衡方法和系统、计算机设备和介质

权利要求书

1.一种样本数量不均衡的数据平衡方法，其特征在于，包括：

S1：根据第一目标检测框从已标注的样本图像中获取第一数量的第一样本图像，按照预设置的第一比例将包含所述第一目标检测框的样本图像划分为训练样本和测试样本；

S2：利用所述训练样本训练预设置的检测模型，利用训练后的检测模型预测所述测试样本并输出检测结果，根据所述检测结果筛选出满足预设置的置信度阈值的第一目标置信框；

S3：按照预设置的第二比例从所述测试样本的各样本图像中随机删除所述第一目标置信框，包括

S31：按照预设置的第二比例从所述测试样本的各样本图像中随机选择待删除的第一目标置信框；

S32：使用所述待删除的第一目标置信框对应的样本图像的各通道均值覆盖所述待删除的第一目标置信框；

S4：根据第二目标检测框从已标注的样本图像中获取第二数量的第二样本图像和该第二样本图像在对应的样本图像中的坐标位置，根据所述第二数量的第二样本图像和坐标位置进行线性回归以获取第二目标检测拟合曲线，其中，所述第一数量大于所述第二数量至少一个数量级，所述第一样本图像和所述第二样本图像为不同类别目标的样本；

S5：根据预设置的扰动阈值对所述第二目标检测拟合曲线进行扰动以获取第二目标检测增强范围；

S6：根据所述第二目标检测框和第二目标检测增强范围随机搜索各样本图像的空白区域，并将所述第二目标检测框随机合成到所述空白区域以生成第二样本增强图像。

2.根据权利要求1所述的数据平衡方法，其特征在于，所述第二目标检测拟合曲线为：

其中，h为线性回归后形成的第二目标检测拟合框的高度，x为所述第二目标检测拟合框距离原点最近点的横坐标值，y为所述距离原点最近点的纵坐标值，w为所述第二目标检测拟合框的宽度，a和b为线性回归拟合的参数。

3.根据权利要求2所述的数据平衡方法，其特征在于，所述第二目标检测增强范围为：

h’＝h±Δα；

其中，h’为扰动后形成的第二目标检测增强框的高度，h为所述第二目标检测拟合框的高度，Δα为预设置的扰动阈值。

4.根据权利要求3所述的数据平衡方法，其特征在于，

在所述S3之后，在所述S6之前，所述数据平衡方法还包括：获取各样本图像中所述第一目标检测框的检测框数量；

所述S6进一步包括：

S61：根据所述第二目标检测框和第二目标检测增强范围从各样本图像中随机搜索所述检测框数量的空白区域；

S62：将所述第二目标检测框随机合成到所述空白区域以生成第二样本增强图像。

5.根据权利要求4所述的数据平衡方法，其特征在于，

所述S61进一步包括：

S611：对所述第二目标检测框进行图像增强变换以获取第二图像增强目标检测框；

S612：根据所述第二图像增强目标检测框和第二目标检测增强范围从各样本图像中随机搜索所述检测框数量的空白区域；

所述S62进一步包括：将所述第二图像增强目标检测框随机合成到所述空白区域以生成第二样本增强图像。

6.根据权利要求5所述的数据平衡方法，其特征在于，所述S611进一步包括：

对所述第二目标检测框进行加噪声、调整图像灰度值、调整图像亮度值中的至少一个以获取第二图像增强目标检测框。