|
|
|
@ -105,7 +105,7 @@ class TestPredictor(CommonTest): |
|
|
|
|
dict_[key], expected_dict[key], rtol=1.e-4, atol=1.e-6) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# @TestPredictor.add_tests |
|
|
|
|
@TestPredictor.add_tests |
|
|
|
|
class TestCDPredictor(TestPredictor): |
|
|
|
|
MODULE = pdrs.tasks.change_detector |
|
|
|
|
TRAINER_NAME_TO_EXPORT_OPTS = { |
|
|
|
@ -177,7 +177,7 @@ class TestCDPredictor(TestPredictor): |
|
|
|
|
self.assertEqual(len(out_multi_array_t), num_inputs) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# @TestPredictor.add_tests |
|
|
|
|
@TestPredictor.add_tests |
|
|
|
|
class TestClasPredictor(TestPredictor): |
|
|
|
|
MODULE = pdrs.tasks.classifier |
|
|
|
|
TRAINER_NAME_TO_EXPORT_OPTS = { |
|
|
|
@ -242,7 +242,7 @@ class TestClasPredictor(TestPredictor): |
|
|
|
|
self.check_dict_equal(out_multi_array_p, out_multi_array_t) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# @TestPredictor.add_tests |
|
|
|
|
@TestPredictor.add_tests |
|
|
|
|
class TestDetPredictor(TestPredictor): |
|
|
|
|
MODULE = pdrs.tasks.object_detector |
|
|
|
|
TRAINER_NAME_TO_EXPORT_OPTS = { |
|
|
|
@ -355,7 +355,7 @@ class TestResPredictor(TestPredictor): |
|
|
|
|
self.assertEqual(len(out_multi_array_t), num_inputs) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# @TestPredictor.add_tests |
|
|
|
|
@TestPredictor.add_tests |
|
|
|
|
class TestSegPredictor(TestPredictor): |
|
|
|
|
MODULE = pdrs.tasks.segmenter |
|
|
|
|
TRAINER_NAME_TO_EXPORT_OPTS = { |
|
|
|
|