def test_inv_trsf(): #--- inverse transformation trsf_inp, trsf_ref = BalTransformation(), BalTransformation() trsf_inp.read(trsf_arbitrary_path) #--- trsf_1 trsf_ref.read(inv_trsf_ref_path) #--- inv(trsf_1) trsf_out = inv_trsf(trsf_inp) ref, out = trsf_ref.mat.to_np_array(), trsf_out.mat.to_np_array() np.testing.assert_array_almost_equal(ref, out, decimal=8) trsf_inp.free() trsf_ref.free() trsf_out.free()
def test_plugin_deformable(self): #--- deformable registration floating_img = imread(data_path('time_0_cut.inr')) reference_img = imread(data_path('time_1_cut.inr')) deformable_trsf = data_path("deformable_0_1.trsf") trsf_inp = BalTransformation() trsf_inp.read(deformable_trsf) trsf_out, res = registration(floating_img, reference_img, method='deformable_registration') np.testing.assert_array_almost_equal(trsf_out.mat.to_np_array(), trsf_inp.mat.to_np_array(), decimal=6) trsf_out.free() trsf_inp.free() return
def test_blockmatching_rigid(self): #--- rigid registration floating_img = imread(data_path('time_0_cut.inr')) reference_img = imread(data_path('time_1_cut.inr')) rigid_trsf = data_path("rigid_0_1.trsf") trsf_out, res = blockmatching(floating_img, reference_img) trsf_inp = BalTransformation() trsf_inp.read(rigid_trsf) np.testing.assert_array_almost_equal(trsf_out.mat.to_np_array(), trsf_inp.mat.to_np_array(), decimal=6) trsf_out.free() trsf_inp.free() return
def test_blockmatching_deformable(self): #--- deformable registration floating_img = imread(data_path('time_0_cut.inr')) reference_img = imread(data_path('time_1_cut.inr')) rigid_trsf = data_path("rigid_0_1.trsf") deformable_trsf = data_path("deformable_0_1.trsf") init_result_transformation = BalTransformation() init_result_transformation.read(rigid_trsf) param_str_2 = '-trsf-type vectorfield' trsf_out, res = blockmatching(floating_img, reference_img, init_result_transformation=init_result_transformation, left_transformation=None, param_str_2=param_str_2) trsf_inp = BalTransformation() trsf_inp.read(deformable_trsf) np.testing.assert_array_almost_equal(trsf_out.mat.to_np_array(), trsf_inp.mat.to_np_array(), decimal=6) trsf_out.free() trsf_inp.free() return
def test_blockmatching_deformable(self): #--- deformable registration floating_img = imread(data_path('time_0_cut.inr')) reference_img = imread(data_path('time_1_cut.inr')) rigid_trsf = data_path("rigid_0_1.trsf") deformable_trsf = data_path("deformable_0_1.trsf") init_result_transformation = BalTransformation() init_result_transformation.read(rigid_trsf) param_str_2 = '-trsf-type vectorfield' trsf_out, res = blockmatching( floating_img, reference_img, init_result_transformation=init_result_transformation, left_transformation=None, param_str_2=param_str_2) trsf_inp = BalTransformation() trsf_inp.read(deformable_trsf) np.testing.assert_array_almost_equal(trsf_out.mat.to_np_array(), trsf_inp.mat.to_np_array(), decimal=6) trsf_out.free() trsf_inp.free() return