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)
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))