c7c80b8429ab70ea3b4d17451eac9cd5020e68a8,dipy/align/tests/test_imwarp.py,,test_em_2d_demons,#,929

Before Change

    matches the saved one.
    """

    moving, static = get_synthetic_warped_circle(1)

    //Configure the metric
    smooth=25.0
    inner_iter=20

After Change

    version of EM in 2D, and this test checks that the current energy profile
    matches the saved one.
    """
    fname = get_data("t1_coronal_slice")
    nslices = 1
    b = 0.1
    m = 4

    image = np.load(fname)
    moving, static = get_warped_stacked_image(image, nslices, b, m)

    //Configure the metric
    smooth=2.0
    inner_iter=20
    q_levels=256
    double_gradient=False
    iter_type="demons"
    metric = metrics.EMMetric(
        2, smooth, inner_iter, q_levels, double_gradient, iter_type)

    //Configure and run the Optimizer
    level_iters = [40, 20, 10]
    optimizer = imwarp.SymmetricDiffeomorphicRegistration(metric, level_iters)
    optimizer.verbosity = VerbosityLevels.DEBUG
    mapping = optimizer.optimize(static, moving, None)
    m = optimizer.get_map()
    assert_equal(mapping, m)

    warped = mapping.transform(moving)
    starting_energy = np.sum((static - moving)**2)
    final_energy = np.sum((static - warped)**2)
    reduced = 1.0 - final_energy/starting_energy

    assert(reduced > 0.9)

Instances