test_common.py 文件源码

def test_scale_enum():
    anchor = numpy.expand_dims(numpy.array([0, 0, 0, 0]), 0)
    scales = numpy.array([1, 2, 3])
    anchor = keras.backend.variable(anchor)
    scales = keras.backend.variable(scales)
    results = keras_rcnn.backend.common._scale_enum(anchor, scales)
    results = keras.backend.eval(results)
    expected = numpy.array(
        [[0, 0, 0, 0], [-0.5, -0.5, 0.5, 0.5], [-1., -1., 1., 1.]])
    numpy.testing.assert_array_equal(results, expected)
    anchor = keras.backend.cast(
        numpy.expand_dims(numpy.array([2, 3, 100, 100]), 0), 'float32')
    anchor = keras.backend.variable(anchor)
    results = keras_rcnn.backend.common._scale_enum(anchor, scales)
    results = keras.backend.eval(results)
    expected = numpy.array([[2., 3., 100., 100.], [-47.5, -46., 149.5, 149.],
                            [-97., -95., 199., 198.]])
    numpy.testing.assert_array_equal(results, expected)