Exemple #1
0
def test_to_decode(boxes_with_label):
    priors = create_prior_boxes('VOC')
    matches = match(boxes_with_label, priors)
    variances = [.1, .2]
    encoded_boxes = encode(matches, priors, variances)
    decoded_boxes = decode(encoded_boxes, priors, variances)
    assert np.all(np.round(decoded_boxes) == matches)
Exemple #2
0
def test_match_box(boxes_with_label, target_unique_matches):
    matched_boxes = match(boxes_with_label, create_prior_boxes('VOC'))
    assert np.array_equal(target_unique_matches,
                          np.unique(matched_boxes[:, :-1], axis=0))