test_deform_conv.py 文件源码

def test_tf_map_coordinates():
    np.random.seed(42)
    input = np.random.random((100, 100))
    coords = np.random.random((200, 2)) * 99

    sp_mapped_vals = map_coordinates(input, coords.T, order=1)
    tf_mapped_vals = tf_map_coordinates(
        K.variable(input), K.variable(coords)
    )
    assert np.allclose(sp_mapped_vals, K.eval(tf_mapped_vals), atol=1e-5)