def fit_ThinPlateSpline_RotReg(x_na, y_ng, bend_coef = .1, rot_coefs = (0.01,0.01,0.0025),scale_coef=.01): import fastrapp f = ThinPlateSpline() rfunc = fastrapp.rot_reg fastrapp.set_coeffs(rot_coefs, scale_coef) f.lin_ag, f.trans_g, f.w_ng = tps_fit_regrot(x_na, y_ng, bend_coef, rfunc) f.x_na = x_na return f
def fit(self, x_na, y_ng, wt_n=None, verbose=True, bend_coef=.1, rot_coef=None): """ x_nd: source cloud y_nd: target cloud wt_n: weight the points """ self.n, self.d = n,d = x_na.shape l_init = self.lin_ag if hasattr(self, "lin_ag") else None self.lin_ag, self.trans_g, self.w_ng = tps.tps_fit_regrot(x_na, y_ng, bend_coef, self.regfunc, wt_n, l_init = l_init, max_iter = self.n_fit_iters, rgrad = self.reggrad) self.x_na = x_na
def fit_ThinPlateSpline_RotReg(x_na, y_ng, bend_coef=.1, rot_coefs=(0.01, 0.01, 0.0025), scale_coef=.01): import fastrapp f = ThinPlateSpline() rfunc = fastrapp.rot_reg fastrapp.set_coeffs(rot_coefs, scale_coef) f.lin_ag, f.trans_g, f.w_ng = tps_fit_regrot(x_na, y_ng, bend_coef, rfunc) f.x_na = x_na return f
def fit(self, x_na, y_ng, wt_n=None, verbose=True, bend_coef=.1, rot_coef=None): """ x_nd: source cloud y_nd: target cloud wt_n: weight the points """ self.n, self.d = n, d = x_na.shape l_init = self.lin_ag if hasattr(self, "lin_ag") else None self.lin_ag, self.trans_g, self.w_ng = tps.tps_fit_regrot( x_na, y_ng, bend_coef, self.regfunc, wt_n, l_init=l_init, max_iter=self.n_fit_iters, rgrad=self.reggrad) self.x_na = x_na