Method test_output

tests/transforms/post/test_probnms.py:65–68

(self, class_args, probs_map, expected)

class TestProbNMS(unittest.TestCase):
    @parameterized.expand(TESTS)
    def test_output(self, class_args, probs_map, expected):
        nms = ProbNMS(**class_args)
        output = nms(probs_map)
        assert_allclose(output, expected)


if __name__ == "__main__":