光条中心线提取-Steger算法(基于Hessian矩阵)类似于ArcScan提取道路中心线

采用结构光进行扫描检测时,需要提取激光条纹的中心线,本文采用经典的Steger算法提取光条中心。

Steger算法原理
Steger算法基于Hessian矩阵,能够实现光条中心亚像素精度定位:首先通过Hessian矩阵能够得到光条的法线方向,然后在法线方向利用泰勒展开得到亚像素位置。
对于图像中激光条纹上的任意一点(x,y)(x,y),Hessian矩阵可以表示为:
H(x,y)=[rxxrxyrxyryy]
H(x,y)=[rxxrxyrxyryy]

其中rxxrxx表示图像沿x的二阶偏导数,其他参数类似。需要注意的是在求Hessian矩阵之前需要对图像进行高斯滤波,高斯滤波时,根据文献[1]中,设置高斯方差σ<w3√σ<w3,其中ww为光条宽度。Hessian矩阵最大特征值对应的特征向量对应于光条的法线方向,用(nx,ny)(nx,ny) 表示,以点(x0,y0)(x0,y0) 为基准点,则光条中心的亚像素坐标为:
(px,py)=(x0+tnx,y0+tny)(px,py)=(x0+tnx,y0+tny)
公式中
t=−nxrx+nyryn2xrxx+2nxnyrxy+n2yryyt=−nxrx+nyrynx2rxx+2nxnyrxy+ny2ryy
如果(tnx,tny)∈[−0.5,0.5]×[−0.5,0.5](tnx,tny)∈[−0.5,0.5]×[−0.5,0.5] ,即一阶导数为零的点位于当前像素内,且(nx,ny)(nx,ny) 方向的二阶导数大于指定的阈值,则该点(x0,y0)(x0,y0) 为光条的中心点,(px,py)(px,py) 则为亚像素坐标。
示例代码
根据Steger算法的原理,借助opencv的Mat数据结构,实现Steger算法如下:
 

void StegerLine()
{ 
    Mat img0 = imread("image_0.png", 1);
    Mat img;
    cvtColor(img0, img0, CV_BGR2GRAY);
    img = img0.clone();

    //高斯滤波
    img.convertTo(img, CV_32FC1);
    GaussianBlur(img, img, Size(0, 0), 6, 6);

    //一阶偏导数
    Mat m1, m2;
    m1 = (Mat_<float>(1, 2) << 1, -1);  //x偏导
    m2 = (Mat_<float>(2, 1) << 1, -1);  //y偏导

    Mat dx, dy;
    filter2D(img, dx, CV_32FC1, m1);
    filter2D(img, dy, CV_32FC1, m2);

    //二阶偏导数
    Mat m3, m4, m5;
    m3 = (Mat_<float>(1, 3) << 1, -2, 1);   //二阶x偏导
    m4 = (Mat_<float>(3, 1) << 1, -2, 1);   //二阶y偏导
    m5 = (Mat_<float>(2, 2) << 1, -1, -1, 1);   //二阶xy偏导

    Mat dxx, dyy, dxy;
    filter2D(img, dxx, CV_32FC1, m3);
    filter2D(img, dyy, CV_32FC1, m4);
    filter2D(img, dxy, CV_32FC1, m5);

    //hessian矩阵
    double maxD = -1;
    int imgcol = img.cols;
    int imgrow = img.rows;
    vector<double> Pt;
    for (int i=0;i<imgcol;i++)
    {
        for (int j=0;j<imgrow;j++)
        {
            if (img0.at<uchar>(j,i)>200)
            {
                Mat hessian(2, 2, CV_32FC1);
                hessian.at<float>(0, 0) = dxx.at<float>(j, i);
                hessian.at<float>(0, 1) = dxy.at<float>(j, i);
                hessian.at<float>(1, 0) = dxy.at<float>(j, i);
                hessian.at<float>(1, 1) = dyy.at<float>(j, i);

                Mat eValue;
                Mat eVectors;
                eigen(hessian, eValue, eVectors);

                double nx, ny;
                double fmaxD = 0;
                if (fabs(eValue.at<float>(0,0))>= fabs(eValue.at<float>(1,0)))  //求特征值最大时对应的特征向量
                {
                    nx = eVectors.at<float>(0, 0);
                    ny = eVectors.at<float>(0, 1);
                    fmaxD = eValue.at<float>(0, 0);
                }
                else
                {
                    nx = eVectors.at<float>(1, 0);
                    ny = eVectors.at<float>(1, 1);
                    fmaxD = eValue.at<float>(1, 0);
                }

                double t = -(nx*dx.at<float>(j, i) + ny*dy.at<float>(j, i)) / (nx*nx*dxx.at<float>(j,i)+2*nx*ny*dxy.at<float>(j,i)+ny*ny*dyy.at<float>(j,i));

                if (fabs(t*nx)<=0.5 && fabs(t*ny)<=0.5)
                {
                    Pt.push_back(i);
                    Pt.push_back(j);
                }
            }
        }
    }

    for (int k = 0;k<Pt.size()/2;k++)
    {
        Point rpt;
        rpt.x = Pt[2 * k + 0];
        rpt.y = Pt[2 * k + 1];
        circle(img0, rpt, 1, Scalar(0, 0, 255));
    }

    imshow("result", img0);
    waitKey(0);
}

示例结果

 

这里写图片描述

参考博客:

其中测绘可能会用到的工具ArcToolBox进行道路中心线提取:ArcMap:矢量线数据提取(道路)中心线

 

  • 2
    点赞
  • 55
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
Steger算法是一种基于Hessian矩阵的边缘检测算法,可以用于亚像素级中心线提取。下面是Python实现的步骤: 1. 导入相关库 ```python import numpy as np from scipy.ndimage import gaussian_filter, maximum_filter from skimage import io, img_as_float ``` 2. 读入图像 ```python img = img_as_float(io.imread('image.jpg', as_gray=True)) ``` 3. 计算Hessian矩阵 ```python sigma = 2 # 高斯滤波参数 img_smooth = gaussian_filter(img, sigma) Ix, Iy = np.gradient(img_smooth) Ixx = gaussian_filter(Ix ** 2, sigma) Ixy = gaussian_filter(Ix * Iy, sigma) Iyy = gaussian_filter(Iy ** 2, sigma) ``` 4. 计算中心线强度 ```python alpha = 0.06 # Hessian矩阵参数 det = Ixx * Iyy - Ixy ** 2 trace = Ixx + Iyy response = det - alpha * trace ** 2 response_max = maximum_filter(response, size=20) response_max[response != response_max] = 0 response_binary = response_max > 0 ``` 5. 中心线提取 ```python def extract_centerline(binary): # 定义八个方向 directions = [(1, 0), (1, 1), (0, 1), (-1, 1), (-1, 0), (-1, -1), (0, -1), (1, -1)] centerline = [] rows, cols = binary.shape for i in range(rows): for j in range(cols): if binary[i, j]: # 计算中心线方向 tangent = np.zeros(2) for d in directions: if i + d[0] >= 0 and i + d[0] < rows and \ j + d[1] >= 0 and j + d[1] < cols and \ binary[i + d[0], j + d[1]]: tangent += d if np.linalg.norm(tangent) > 0: tangent /= np.linalg.norm(tangent) centerline.append((j, i, tangent[0], tangent[1])) return centerline centerline = extract_centerline(response_binary) ``` 6. 输出中心线线方程 ```python for line in centerline: x, y, tx, ty = line a = -ty / tx b = 1 c = -(a * x + b * y) print('Line equation: {}x + {}y + {} = 0'.format(a, b, c)) ``` 这样就可以得到中心线的直线方程。需要注意的是,在实际应用中,可能需要对中心线进行进一步的筛选和优化。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值