[发明专利]目标检测数据集生成方法、装置、设备及存储介质

权利要求书

1.一种目标检测数据集生成方法，其特征在于，所述目标检测数据集生成方法包括：

遍历预设背景图文件夹路径，获取遍历到的当前背景图；

根据所述当前背景图对应的文件名生成空白标注文件；

将所述当前背景图以及所述空白标注文件存储至初始数据集中，得到目标数据集。

2.如权利要求1所述的目标检测数据集生成方法，其特征在于，所述将所述当前背景图以及所述空白标注文件存储至初始数据集中，得到目标数据集之后，所述方法还包括：

根据所述目标数据集对预设目标检测模型进行训练，得到训练好的目标检测模型。

3.如权利要求2所述的目标检测数据集生成方法，其特征在于，所述根据所述目标数据集对预设目标检测模型进行训练，得到训练好的目标检测模型之前，所述方法还包括：

确定训练代码对应的文件读取格式；

根据所述文件读取格式对所述目标数据集中的所述空白标注文件进行格式转换，得到调整后的目标数据集；

所述根据所述目标数据集对预设目标检测模型进行训练，得到训练好的目标检测模型，包括：

根据所述调整后的目标数据集对预设目标检测模型进行训练，得到训练好的目标检测模型。

4.如权利要求3所述的目标检测数据集生成方法，其特征在于，所述根据所述调整后的目标数据集对预设目标检测模型进行训练，得到训练好的目标检测模型，包括：

加载所述调整后的目标数据集，获取所述调整后的目标数据集中的训练图像以及训练标注文件；

根据所述训练图像生成模型输入特征；

根据所述训练标注文件生成模型预测结果；

根据所述模型输入特征以及所述模型预测结果对预设目标检测模型进行训练，得到训练好的目标检测模型。

5.如权利要求1所述的目标检测数据集生成方法，其特征在于，所述将所述当前背景图以及所述空白标注文件存储至初始数据集中，得到目标数据集之前，所述方法还包括：

获取各类别分别对应的多个正样本图像；

根据所述正样本图像以及所述正样本图像对应的标注文件生成初始数据集。

6.如权利要求1所述的目标检测数据集生成方法，其特征在于，所述遍历预设背景图文件夹路径，获取遍历到的当前背景图之前，所述方法还包括：

从预设数据集中搜索包含误识别标签的若干样本图像；

根据所述若干样本图像对应的标注文件分别获取各样本图像对应的检测框信息；

根据所述检测框信息从所述若干样本图像中分别截取对应的区域，得到若干初始图像；

从预设背景图文件夹路径获取初始背景图；

将所述若干初始图像分别与所述初始背景图进行融合，得到若干融合背景图；

分别对所述若干融合背景图进行命名，并将所述若干融合背景图存储于所述预设背景图文件夹路径。

7.如权利要求1所述的目标检测数据集生成方法，其特征在于，所述遍历预设背景图文件夹路径，获取遍历到的当前背景图之前，所述方法还包括：

根据训练好的第一目标检测模型对若干待识别的图像进行识别；

在识别到目标图像为背景图时，将所述目标图像存储于预设背景图文件夹路径。

8.一种目标检测数据集生成装置，其特征在于，所述目标检测数据集生成装置包括：

获取模块，用于遍历预设背景图文件夹路径，获取遍历到的当前背景图；

标注文件生成模块，用于根据所述当前背景图对应的文件名生成空白标注文件；

数据集生成模块，用于将所述当前背景图以及所述空白标注文件存储至初始数据集中，得到目标数据集。

9.一种目标检测数据集生成设备，其特征在于，所述设备包括：存储器、处理器及存储在所述存储器上并可在所述处理器上运行的目标检测数据集生成程序，所述目标检测数据集生成程序配置为实现如权利要求1至7中任一项所述的目标检测数据集生成方法。

10.一种存储介质，其特征在于，所述存储介质上存储有目标检测数据集生成程序，所述目标检测数据集生成程序被处理器执行时实现如权利要求1至7任一项所述的目标检测数据集生成方法。