2.移动最小二乘法
转载自:基于移动最小二乘法的曲线曲面拟合--一碗竹叶青
如果离散数据量比较大,形状复杂,用传统最小二乘法会很奇怪。
所以又有了改良型的基于移动最小二乘法。
区别:
( 1)拟合函数的建立不同。这种方法建立拟合函数不是采用传统的多项式或其它函数,而是由一个系数向量 a(x)和基函数 p(x)构成, 这里 a(x)不是常数,而是坐标 x 的函数。
( 2)引入紧支( Compact Support)概念,认为点 x 处的值 y 只受 x
附近子域内节点影响,这个子域称作点 x 的影响区域, 影响区域外的节点对 x的取值没有影响。在影响区域上定义一个权函数w(x), 如果权函数在整个区域取为常数, 就得到传统的最小二乘法。参考自《基于移动最小二乘法的曲线曲面拟合-曾清红》
1.拟合函数的建立
在拟合区域的一个局部子域上, 拟合函数 f (x)表示为:式中
这个基函数我还真的看不懂,但大家都是这样设的
在移动最小二乘近似中, 系数 ai(x) 是通过令近似函数 u(x) 在点 x 的邻域内各节点误差的加权平方和为最小来确定的式中 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()