Пример #1
0
def test_unnormalized_vlad():
    features = np.array([[0, 1.1]])
    centers = np.array([
        [1, 0],
        [0, 1],
    ])

    res = vlad.unnormalized_vlad(features, centers)

    assert res[0] == res[1] == res[2] == 0
    assert pytest.approx(res[3], 1e-6) == 0.1
Пример #2
0
def test_unnormalized_vlad() -> None:
    features = np.array([[0, 1.1]])
    centers = np.array([
        [1.0, 0.0],
        [0.0, 1.0],
    ])

    res = vlad.unnormalized_vlad(features, centers)

    # pyre-fixme[16]: Optional type has no attribute `__getitem__`.
    assert res[0] == res[1] == res[2] == 0
    assert pytest.approx(res[3], 1e-6) == 0.1