[发明专利]一种图像结构边缘的定位方法及装置

权利要求书

1.一种图像结构边缘的定位方法，其特征在于，所述方法包括：

获取待检测图像；

采用预先训练好的边缘检测模型，对所述待检测图像进行边缘检测，得到边缘得分矩阵；

对所述边缘得分矩阵进行直线检测；

根据所述直线检测结果确定候选结构边缘；

根据预设的约束条件，在所述候选结构边缘中确定目标结构边缘；

其中，训练所述边缘检测模型的步骤为：获取图像样本；

对所述图像样本进行标注，确定以结构边缘像素点为中心的图像块为正样本，以非结构边缘像素点为中心的图像块为负样本；

获取所述正样本和负样本的多通道特征，所述多通道特征由颜色通道特征，梯度幅值通道特征和梯度方向通道特征组成，所述颜色通道特征为ycbcr或lab颜色通道特征；

根据所述图像样本，以及所述正样本和负样本的多通道特征，对机器学习分类器进行训练得到边缘检测模型。

2.根据权利要求1所述的方法，其特征在于，所述根据所述直线检测结果确定候选结构边缘的步骤，包括：

根据所述直线检测结果，确定潜在直线；

根据所述潜在直线的角度，确定水平线和垂直线；

根据所述水平线和所述垂直线，获得候选结构边缘。

3.根据权利要求2所述的方法，其特征在于，所述根据所述水平线和所述垂直线，获得候选结构边缘的步骤，包括：

根据任意两条所述水平线和任意两条所述垂直线，获得候选结构边缘，所述候选结构边缘的形状为四边形。

4.一种图像结构边缘的定位装置，其特征在于，所述装置包括：

图像获取模块，用于获取待检测图像；

边缘检测模块，用于采用预先训练好的边缘检测模型，对所述待检测图像进行边缘检测，得到边缘得分矩阵；

直线检测模块，用于对所述边缘得分矩阵进行直线检测；

边缘候选模块，用于根据所述直线检测模块的结果确定候选结构边缘；

边缘确定模块，用于根据预设的约束条件，在所述候选结构边缘中确定目标结构边缘；

所述装置还包括模型训练模块，所述模型训练模块包括：

样本获取子模块，用于获取图像样本；

样本标注子模块，用于对所述图像样本进行标注，确定以结构边缘像素点为中心的图像块为正样本，以非结构边缘像素点为中心的图像块为负样本；

特征获取子模块，用于获取所述正样本和负样本的多通道特征；所述多通道特征由颜色通道特征，梯度幅值通道特征和梯度方向通道特征组成，所述颜色通道特征为ycbcr或lab颜色通道特征；

模型获得子模块，用于根据所述图像样本，以及所述正样本和负样本的多通道特征，对机器学习分类器进行训练得到边缘检测模型。

5.根据权利要求4所述的装置，其特征在于，所述边缘候选模块，包括：

直线确定子模块，用于根据所述直线检测模块的结果，确定潜在直线；

直线分类子模块，用于根据所述潜在直线的角度，确定水平线和垂直线；

边缘构成子模块，用于根据所述水平线和所述垂直线，获得候选结构边缘。

6.根据权利要求5所述的装置，其特征在于，所述边缘构成子模块，包括：

四边形构成子模块，用于根据任意两条所述水平线和任意两条所述垂直线，获得候选结构边缘，所述候选结构边缘的形状为四边形。