39ee36a7c2c9cd8e04cfbb19c64e55542f76c794,tests/layers/object_detection/test_proposal_target.py,,test_get_bbox_regression_labels,#,251

Before Change

def test_get_bbox_regression_labels():
    n = 10
    bbox_target_data = keras.backend.zeros((n, 4))
    num_classes = 3
    labels = numpy.reshape(
        [[0, 1, 0], [1, 0, 0], [1, 0, 0], [0, 0, 1], [0, 1, 0]], (1, -1, 3))

After Change

    bbox_target_data = keras.backend.variable(bbox_target_data)
    num_classes = 3
    labels = numpy.array([[0, 1, 0], [1, 0, 0], [1, 0, 0], [0, 0, 1], [0, 1, 0]])
    labels = numpy.argmax(labels, axis=1)
    labels = keras.backend.variable(labels)
    bbox_targets = proposal_target.get_bbox_regression_labels(bbox_target_data, labels, num_classes)
    bbox_targets = keras.backend.eval(bbox_targets)

Instances