test_common.py 文件源码

def test_whctrs():
    anchor = keras.backend.cast(keras.backend.expand_dims([0, 0, 0, 0], 0),
                                'float32')
    results0, results1, results2, results3 = keras_rcnn.backend.common._whctrs(
        anchor)
    results = numpy.array(
        [keras.backend.eval(results0), keras.backend.eval(results1),
         keras.backend.eval(results2), keras.backend.eval(results3)])
    expected = numpy.expand_dims([1, 1, 0, 0], 1)
    numpy.testing.assert_array_equal(results, expected)
    anchor = keras.backend.cast(keras.backend.expand_dims([2, 3, 100, 100], 0),
                                'float32')
    results0, results1, results2, results3 = keras_rcnn.backend.common._whctrs(
        anchor)
    results = numpy.array(
        [keras.backend.eval(results0), keras.backend.eval(results1),
         keras.backend.eval(results2), keras.backend.eval(results3)])
    expected = numpy.expand_dims([99, 98, 51, 51.5], 1)
    numpy.testing.assert_array_equal(results, expected)