基于移动最小二乘的图像变形和曲线拟合

在最近的项目中经常遇到给出几个点需要拟合出一条曲线。

在离散的点云中,求曲线曲面拟合,不能简单地连接这些点,如果知道曲线曲面的形式,如为二次曲线等,可以简单地使用最小二乘法估计参数;但如果曲线曲面形式未知,可以使用移动最小二乘法或者主曲线方法。

MLS的讲解

移动最小二乘法是在最小二乘法基础上加以改进的,添加了权函数等,具体的可以参考论文,“移动最小二乘法论文”链接,这篇论文对MLS讲解的很详细,最后还给出了程序设计思路。

我做一点点说明,论文中的矩阵A的写法欠妥,其他关于移动最小二乘法研究中还有另外一种写法:这里写图片描述,这里的B对应论文中的P,这点要注意,这样的话A就是一个矩阵。如果是线性基的二维曲线,矩阵A就是: 

这里写图片描述,依次类推,其他的可以详看论文。

MLS代码块

代码的话我是根据论文中提供的程序设计,再结合一些网上的资料编写出来的,编程语言是C++;当然我也编写了python,应该是先编python,再编的C++。原因是python中可以加载一个矩阵运算库,C++中没有矩阵运算,要自己编写库,大家可以参考我这篇博客,介绍了矩阵运算链接。但是后来实验发现,python跑起来很费时间,C++只需它的一半的时间久跑完了,需要python代码也可以私信我,这里就不贴了。哦,对了,代码是包含很多自定义函数和变量,大家不要瞎贴代码,对照那篇论文的程序设计思路一下子就懂了,话不多说,上代码:

//移动最小二乘法的具体计算过程,参照论文“基于移动最小二乘法的曲线曲面拟合”,AB矩阵参照论文“移动最小二乘法的研究”
int MLS_Calc(int x_val,int y_val,float x[],float y[],float z[])
{
    int max_delta=max_x-min_x;//区域半径
    float p[M][N]={0};
    float sumf[N][N]={0};
    float w[M]={0};
    for(int j=0;j<M;j++)//求w
    {
        float s=fabs((x[j]-x_val))/max_delta;
        if(s<=0.5)
            w[j]=2/3.0-4*s*s+4*s*s*s;
        else
        {
            if(s<=1)
                w[j]=4/3.0-4*s+4*s*s-4*s*s*s/3.0;
            else
                w[j]=0;
        }
        p[j][0]=1;//每个采样点计算基函数
        p[j][1]=x[j];
        p[j][2]=y[j];
        p[j][3]=x[j]*x[j];
        p[j][4]=x[j]*y[j];
        p[j][5]=y[j]*y[j];
    }
    f(w,x,y,sumf,p);//计算得出A矩阵

    float p1[N];
    Matrix A=Trans_Matrix(sumf,N);
    Matrix A_1=m_c.Matrix_copy(&A);
    m_c.Matrix_inv(&A_1);//求A矩阵的逆A_1

    Matrix B(N,1);//求矩阵B,N行M列
    B.init_Matrix();
    for(int j=0;j<M;j++)//求得B矩阵的每列
    {
        p1[0]=1*w[j];
        p1[1]=x[j]*w[j];
        p1[2]=y[j]*w[j];
        p1[3]=x[j]*x[j]*w[j];
        p1[4]=x[j]*y[j]*w[j];
        p1[5]=y[j]*y[j]*w[j];
        Matrix P=Trans_Matrix_One(p1,N);//数组P1转成1行N列的P矩阵
        if(j==0)//第一列直接赋值
        {
            for(int i=0;i<N;i++)
                B.write(i,0,p1[i]);
        }
        else
        {
            m_c.Matrix_trans(&P);//矩阵转置,P转为N行1列矩阵
            m_c.Matrix_addCols(&B,&P);//矩阵B列附加,形成N行M列矩阵
        }
        P.free_Matrix();
    }

    float D[N]={1,x_val,y_val,x_val*x_val,x_val*y_val,y_val*y_val};
    Matrix D1=Trans_Matrix_One(D,N);//转成1行N列矩阵

    Matrix D_A1_mul(1,N);//定义矩阵并初始化相乘的结果矩阵,1行N列
    D_A1_mul.init_Matrix();
    if(m_c.Matrix_mul(&D1,&A_1,&D_A1_mul)==-1)
        cout<<"矩阵有误1!";//1行N列矩阵乘以N行N列矩阵得到结果为1行N列

    Matrix D_A1_B_mul(1,M);//定义矩阵并初始化相乘的结果矩阵,1行M列
    D_A1_B_mul.init_Matrix();
    if(m_c.Matrix_mul(&D_A1_mul,&B,&D_A1_B_mul)==-1)
        cout<<"矩阵有误2";//1行N列矩阵乘以N行M列矩阵得到记过矩阵为1行M列

    Matrix z1=Trans_Matrix_One(z,M);//将数组z转换成1行M列矩阵
    m_c.Matrix_trans(&z1);//转置得到M行1列矩阵
    Matrix Z(1,1);//得到矩阵结果,1行1列
    Z.init_Matrix();
    if(m_c.Matrix_mul(&D_A1_B_mul,&z1,&Z)==-1)
        cout<<"矩阵有误3!";//1行M列矩阵乘以M行1列矩阵得到1行1列矩阵,即值Z

    float z_val=Z.read(0,0);
    if(z_val>255)
        z_val=255;
    if(z_val<0)
        z_val=0;

    A.free_Matrix();
    A_1.free_Matrix();
    B.free_Matrix();

    D1.free_Matrix();
    D_A1_mul.free_Matrix();
    D_A1_B_mul.free_Matrix();
    z1.free_Matrix();
    Z.free_Matrix();

    return (int)z_val;
}

 

 

 

 

 

posted @ 2018-06-28 09:52  Anita-ff  阅读(1878)  评论(0编辑  收藏  举报