[发明专利]一种训练样本图像获取方法及装置

权利要求书

1.一种训练样本图像获取方法，其特征在于，包括以下步骤：

获取待处理大尺度训练样本图像；

确定训练样本图像中所有缺陷的位置和大小；

根据每个缺陷的位置和大小计算待剪切缺陷的需要剪切的小图像个数；

根据所述待剪切缺陷的需要剪切的小图像个数，对待剪切缺陷进行剪切，将剪切完成的缺陷标记为已剪切的缺陷；

复制训练样本图像中已标记为已剪切的缺陷的位置区域，得到具有缺陷的小尺寸训练样本图像。

2.根据权利要求1所述的训练样本图像获取方法，其特征在于，所述确定大尺度训练样本图像中所有缺陷的位置和大小，包括：

对所述大尺度训练样本图像中的缺陷进行blob分析，并对每个缺陷所处外接矩形进行网格化处理，获取大尺度训练样本图像中的缺陷在所述外接矩形中的位置和大小。

3.根据权利要求2所述的训练样本图像获取方法，其特征在于，并对每个缺陷所处外接矩形进行网格化处理，包括:

并根据每个缺陷的blob的质心位置、每个缺陷所处外接矩形的大小及需要剪切的小图像的大小对每个外接矩形进行网格化处理。

4.根据权利要求2或3所述的训练样本图像获取方法，其特征在于，所述根据每个缺陷的位置和大小计算待剪切缺陷的需要剪切的小图像个数，包括：

查找所有待剪切的缺陷；

根据所述缺陷外接矩形的大小、位置及需要剪切的小图像的大小计算每个待剪切缺陷中需要剪切的小图个数。

5.根据权利要求4所述的训练样本图像获取方法，其特征在于，将待剪切缺陷进行剪切之前，还包括：

过滤掉待剪切缺陷所处矩形网格中无缺陷的矩形网格。

6.根据权利要求5所述的训练样本图像获取方法，其特征在于，所述过滤掉待剪切缺陷所处矩形网格中无缺陷的矩形网格之后，还包括：

对有缺陷的矩形网格内的缺陷进行剪切，将剪切完成的缺陷标记为已剪切的缺陷。

7.根据权利要求6所述的训练样本图像获取方法，其特征在于，当两个待剪切缺陷位于同一个矩形网格，对其中一个待剪切缺陷进行剪切时，另一个待剪切缺陷也被剪切，并标记为已剪切的缺陷。

8.一种训练样本图像获取装置，其特征在于，包括：

Blob分析模块，用于对所述大尺度训练样本图像中的缺陷进行blob分析，并对每个缺陷所处外接矩形进行网格化处理，获取大尺度训练样本图像中的缺陷在所述外接矩形中的位置和大小；

剪切位置信息生成模块，用于根据每个缺陷的位置和大小计算待剪切缺陷的需要剪切的小图像个数，将所述待剪切缺陷进行剪切，将剪切完成的缺陷标记为已剪切，将将剪切完成的缺陷标记为已剪切的缺陷；

小尺寸缺陷图像生成模块，用于复制标记为已剪切的缺陷的位置区域，得到具有缺陷的小尺寸图像。

9.根据权利要求8所述的训练样本图像获取装置，其特征在于，所述剪切位置信息生成模块包括：

计算单元,所述计算单元用于查找所有待剪切的缺陷，根据所述缺陷外接矩形的大小、位置及需要剪切的小图像的大小计算每个待剪切缺陷中需要剪切的小图个数；

过滤单元，所述过滤单元用于过滤待剪切缺陷所处矩形网格中无缺陷的矩形网格。