[发明专利]样本数据生成方法及装置、训练模型的方法及装置

权利要求书

1.一种样本数据生成方法，其特征在于，所述方法包括：

获取样本图片，所述样本图片中包含多个目标类别；

确定所述样本图片中分布占比小于第一预设占比的第一目标类别，和/或分布占比大于第二预设占比的第二目标类别；

按照预设窗口大小对所述样本图片进行遍历，生成待分析切片；

按照预设筛选条件从待分析切片中确定样本数据，使得得到的样本数据符合如下条件：

针对以第一目标类别为筛选依据的情况，样本数据中包含第一目标类别的样本数据比例增多；针对以第二目标类别为筛选依据的情况，样本数据中包含第二目标类别的样本数据比例减少；

所述样本数据生成方法，还包括：

对所述样本图片中的像素进行标注，得到标注图；其中，构成同一目标类别的像素具有相同的标注值；

根据所述标注图中不同目标类别对应的不同标注值，确定各目标类别在所述样本图片中的分布占比；

针对样本图片数量大于一的情况，根据所述标注图中不同目标类别对应的不同标注值，确定各目标类别在所述样本图片中的分布占比，包括：

针对不同目标类别，将该目标类别对应标注值在各样本图片分别对应的标注图中的总数量，与各样本图片分别对应的标注图中像素总数量的比值，确定为该目标类别在样本图片中的分布占比；

针对以第一目标类别为筛选依据的情况，按照预设筛选条件从待分析切片中确定样本数据，包括：

确定待分析切片中所包含的第一目标类别在该待分析切片中的分布占比；

若确定的分布占比大于第一预设切片占比，则将该待分析切片确定为样本数据，若确定出分布占比小于或等于第一预设切片占比，舍弃该待分析切片；其中，所述第一预设切片占比是基于所述第一预设占比确定的；

针对以第二目标类别为筛选依据的情况，按照预设筛选条件从待分析切片中确定样本数据，包括：

确定待分析切片中所包含的第二目标类别在该待分析切片中的分布占比；

若确定的分布占比小于第二预设切片占比，则将该待分析切片确定为样本数据，若确定出分布占比大于或等于第二预设切片占比，舍弃该待分析切片，其中，所述第二预设切片占比是基于所述第二预设占比确定的。

2.根据权利要求1所述的方法，其特征在于，按照预设筛选条件从待分析切片中确定样本数据，包括：

针对按照指定预设窗口大小对所述样本图片进行遍历所生成的待分析切片，按照预设筛选条件从待分析切片中确定样本数据。

3.根据权利要求2所述的方法，其特征在于，第一目标类别在所述样本图片中的分布占比和第二目标类别在所述样本图片中的分布占比之间的差值越大，指定预设窗口大小数量越多。

4.一种样本数据训练模型的方法，其特征在于，所述方法包括：

确定样本数据的标注样本数据，其中，所述样本数据是基于如权利要求1-3任一项所述的样本数据生成方法生成的；

将样本数据以及标注样本数据输入语义分割模型对所述语义分割模型进行训练；其中，所述语义分割模型用于实现语义分割。