Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-18 17:27:43 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933]enable image segmentation test

Browse Source 
enable image segmentation test
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9522337
This commit is contained in:
hejunjie.hjj
2022-07-26 17:49:20 +08:00
committed by yingda.chen
parent 5985bd10ba
commit c112e599fe
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

2
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,