Browse Source

[to #42322933]enable image segmentation test

enable image segmentation test
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9522337
master
hejunjie.hjj yingda.chen 3 years ago
parent
commit
c112e599fe
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      tests/pipelines/test_image_instance_segmentation.py
  2. +1
    -1
      tests/trainers/test_image_instance_segmentation_trainer.py

+ 2
- 2
tests/pipelines/test_image_instance_segmentation.py View File

@@ -18,7 +18,7 @@ class ImageInstanceSegmentationTest(unittest.TestCase):
model_id = 'damo/cv_swin-b_image-instance-segmentation_coco'
image = 'data/test/images/image_instance_segmentation.jpg'

@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_run_with_model_from_modelhub(self):
model = Model.from_pretrained(self.model_id)
config_path = os.path.join(model.model_dir, ModelFile.CONFIGURATION)
@@ -30,7 +30,7 @@ class ImageInstanceSegmentationTest(unittest.TestCase):
preprocessor=preprocessor)
print(pipeline_ins(input=self.image)[OutputKeys.LABELS])

@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
def test_run_with_model_name(self):
pipeline_ins = pipeline(
task=Tasks.image_segmentation, model=self.model_id)


+ 1
- 1
tests/trainers/test_image_instance_segmentation_trainer.py View File

@@ -71,7 +71,7 @@ class TestImageInstanceSegmentationTrainer(unittest.TestCase):
shutil.rmtree(self.tmp_dir)
super().tearDown()

@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_trainer(self):
kwargs = dict(
model=self.model_id,


Loading…
Cancel
Save