Преглед изворни кода

Merge branch 'update_docs' of https://github.com/Bobholamovic/PaddleRS into update_docs

Bobholamovic пре 3 година
родитељ
комит
b3e0c2eaf0
2 измењених фајлова са 4 додато и 4 уклоњено
  1. BIN
      docs/images/whole_picture.png
  2. 4 4
      tests/deploy/test_predictor.py

BIN
docs/images/whole_picture.png


+ 4 - 4
tests/deploy/test_predictor.py

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