移动最小二乘法

2.移动最小二乘法

转载自:基于移动最小二乘法的曲线曲面拟合--一碗竹叶青

如果离散数据量比较大,形状复杂,用传统最小二乘法会很奇怪。
所以又有了改良型的基于移动最小二乘法。
区别:
( 1)拟合函数的建立不同。这种方法建立拟合函数不是采用传统的多项式或其它函数,而是由一个系数向量 a(x)和基函数 p(x)构成, 这里 a(x)不是常数,而是坐标 x 的函数。
( 2)引入紧支( Compact Support)概念,认为点 x 处的值 y 只受 x
附近子域内节点影响,这个子域称作点 x 的影响区域, 影响区域外的节点对 x的取值没有影响。在影响区域上定义一个权函数w(x), 如果权函数在整个区域取为常数, 就得到传统的最小二乘法。参考自《基于移动最小二乘法的曲线曲面拟合-曾清红》

1.拟合函数的建立

在拟合区域的一个局部子域上, 拟合函数 f (x)表示为:
拟合函数

式中
为待求系数,它是坐标x的函数。
称为基函数。它是一个k阶完备的多项式,m是基函数的项数

对于一维问题,基函数可以为
一维基函数
二维基函数

这个基函数我还真的看不懂,但大家都是这样设的

在移动最小二乘近似中, 系数 ai(x) 是通过令近似函数 u(x) 在点 x 的邻域内各节点误差的加权平方和为最小来确定的

对a(x)求偏导,令其等于零就可以得到及极少值。
令对a(x)偏导为0

式中 n 为点 x 的邻域 内所包含的节点数.,w(x)=(x−xI) 称为节点 xI 处的权函数, 它在节点 xI 周围的一个有限区域中大于零, 而在该区域外为零 . 权函数的定义表明, 只有在节点 xI 的影响域范围内的节点才对该点的近似函数产生影响.

2.支撑域:


将整个x范围划分为若干个区域,每个区域包含若干个x点,那么并且规定其中一点为标准点,其他点为参考点。
参考点与标准点的距离作为权函数的参数。得出权重。

3.权函数:

!
权函数(样条函数)

4.法方程的推导:
对于任意函数 h(x) 和 g(x), 引入记号:



那么,公式4可写成


->

写成矩阵形式:写成矩阵形式:

由上面的法方程, 解得 a(x).
然后求解得出A(x),B(x) 求解得出α(x)


5.拟合流程


6.网格化
这里说明一下为什么要网格化,网格化主要是选取标准点,并以标准点来划分支撑域,确定支撑域半径和支撑域内的节点 x

7.python代码

#主题部分
X=np.arange(-0.9,0.9,0.05)
# 数据点x个数
M=len(x)
# 基函数个数
N=2
p=np.zeros((M,2))
Y=[]
for XX in X:
    w = np.zeros((M,1))
    d=0.1 # 影响区域的半径
    for i in range(0,M):
        w[i]=W_fun(d, x[i], XX)
        p[i][0]=1
        p[i][1]=x[i]
    A=fun_A(x,w,p)
    B=fun_B(y,w,p)
    a=np.linalg.solve(A,B)
    Y.append(a[0]+a[1]*XX)

#其他函数部分
# 权函数
def W_fun(d,x,X):
    s=abs(x-X)/d
    if (s<=0.5):
        return (2/3)-4*s**2+4*s**3
    elif(s<=1):
        return (4/3)-4*s+4*s**2-(4/3)*s**3
    else:
        return 0
# 权函数记号(pm,pn)的计算
def pm_pn(w,x,p,m,n):
    # x为数据点,w为支撑域的权重,M为数据点个数 p1,p2为传入的数值
    pmn=0
    M=len(x)
    # i代表数据点,m n代表(pm,pn)的下标
    for i in range(M):
        pmn=pmn+w[i]*p[i][m]*p[i][n]
    return float(pmn)
# B矩阵的建立
def fun_B(u,w,p):
    pumI=0
    M=len(u) #数据点个数
    m=p.shape[1] # 基函数个数
    B=[]
    for j in range(m):
        for i in range(M):
            pumI=pumI+w[i]*p[i][j]*u[i]
        B.append(float(pumI))
    return B
 # A矩阵的建立
def fun_A(x,w,p):
    M=len(x)函数
    m=p.shape[1]
    A=[]
    for mm in range(m):
        matA=[]
        for nn in range(m):
            pmn=pm_pn(w,x,p,mm,nn)
            matA.append(pmn)
        A.append(matA)
    return A
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Tue Apr  2 11:39:26 2019

@author: yxh
"""
# Python实现MLS曲线拟合
import time
import numpy as np
from matplotlib import pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
#其他函数部分
# 权函数
def W_fun(d,x,X):
    s=abs(x-X)/d
    if (s<=0.5):
        return (2/3)-4*s**2+4*s**3
    elif(s<=1):
        return (4/3)-4*s+4*s**2-(4/3)*s**3
    else:
        return 0
# 权函数记号(pm,pn)的计算
def pm_pn(w,x,p,m,n):
    # x为数据点,w为支撑域的权重,M为数据点个数 p1,p2为传入的数值
    pmn=0
    M=len(x)
    # i代表数据点,m n代表(pm,pn)的下标
    for i in range(M):
        pmn=pmn+w[i]*p[i][m]*p[i][n]
    return float(pmn)
# B矩阵的建立
def fun_B(u,w,p):
    pumI=0
    M=len(u) # 数据点个数
    m=p.shape[1] # 基函数个数
    B=[]
    for j in range(m):
        for i in range(M):
            pumI=pumI+w[i]*p[i][j]*u[i]
        B.append(float(pumI))
    return B

def fun_A(x,w,p):
    M=len(x)
    m=p.shape[1]
    A=[]
    for mm in range(m):
        matA=[]
        for nn in range(m):
            pmn=pm_pn(w,x,p,mm,nn)
            matA.append(pmn)
        A.append(matA)
    return A



path = '/home/yxh/vispy/001095_curve.txt'
pos = np.loadtxt(path)
pos_rgb = pos[:, -1]
idxs = np.where(pos_rgb == 1)
lane = pos[idxs]
x = lane[:, 0]
y = lane[:, 1]
z = lane[:, 2]
lane = lane[lane[:, 0].argsort()]
nums = 0
num = np.uint8(lane.shape[0]/50) + 1
lane_nums = np.zeros((num, 3))
for i in range(lane.shape[0]):
    if(i==0 or i%50==0):
        lane_nums[nums, :] = lane[i, 0:3]
        nums += 1

X = lane_nums[:, 0]
#Y = lane_nums[:, 1]
Z = lane_nums[:, 2]
#X=np.arange(np.min(x, axis = 0), np.max(x, axis = 0), 1.0)
# 数据点x个数
M=len(x)
# 基函数个数
N=6
p=np.zeros((M,N))
Y=[]
time_begin = time.time()
ids = 0
for XX in X:
    w = np.zeros((M,1))
    d=4.0 # 影响区域的半径
    for i in range(0,M):
        w[i]=W_fun(d, x[i], XX)
        p[i][0]=1
        p[i][1]=x[i]
        p[i][2]=z[i]
        p[i][3]=pow(x[i], 2)
        p[i][4]=x[i]*z[i]
        p[i][5]=pow(z[i], 2)
    A=fun_A(x,w,p)
    B=fun_B(y,w,p)
    print len(A), len(B)
    a=np.linalg.solve(A,B)
    Y.append(a[0]+a[1]*XX+a[2]*Z[ids]+a[3]*pow(XX, 2)+a[4]*XX*Z[ids]+a[5]*pow(Z[ids], 2))
    ids += 1
time_end = time.time()
print 'time_cost:', time_end-time_begin
Z = np.zeros((X.shape))
print Z, Z.shape
f1 = np.polyfit(X, Y, 2)   # Least squares polynomial fit. use 4 can get best fitting
p1 = np.poly1d(f1) 
print 'xishu:', f1
print 'fangcheng:', p1
yval = p1(X)
final_pos = np.zeros((len(Y), 3))
final_pos[:, 0] = X
final_pos[:, 1] = yval
final_pos[:, 2] = np.mean(z, axis = 0)
#Z = final_pos[:, 2]
print final_pos
print ">>>>>>>>>>>>>>>>>final_pos[0]:", final_pos[0]
print ">>>>>>>>>>>>>>>>>final_pos[-1]:", final_pos[-1]
print final_pos.shape
"""
plot1 = plt.plot(X, Y, '-s', label='original values')
plot2 = plt.plot(X, yval, 'r', label='ployfit values')
plt.xlabel('x')
plt.ylabel('y')
plt.legend(loc=4)
plt.title('ployfitting')
plt.show()
"""

fig = plt.Figure()
ax = plt.gca(projection='3d')
#ax3 = plt.axes(projection='3d')
ax.set_title('3D-curve')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
figure1 = ax.plot(X, Y, Z, c = 'r')  #red
figure2 = ax.plot(X, yval, Z)   #blue
plt.show()

#xx, yy = np.meshgrid(X, Y)
#print xx.shape

#zz = np.zeros((xx.shape))
#print zz.shape

#ax3.plot_surface(xx, yy, zz, cmap='rainbow')
#plt.show()

你可能感兴趣的:(移动最小二乘法)