This commit is contained in:
xingjun.wang
2023-05-22 10:53:18 +08:00
parent 52aea36c12
commit 48c0d2a9af
468 changed files with 12942 additions and 7176 deletions

View File

@@ -9,11 +9,10 @@ from modelscope.outputs import OutputKeys
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.utils.cv.image_utils import semantic_seg_masks_to_image
from modelscope.utils.demo_utils import DemoCompatibilityCheck
from modelscope.utils.test_utils import test_level
class ImageSemanticSegmentationTest(unittest.TestCase, DemoCompatibilityCheck):
class ImageSemanticSegmentationTest(unittest.TestCase):
def setUp(self) -> None:
self.task = 'image-segmentation'
@@ -54,10 +53,6 @@ class ImageSemanticSegmentationTest(unittest.TestCase, DemoCompatibilityCheck):
cv2.imwrite('result.jpg', draw_img)
print('test_image_semantic_segmentation_vitadapter_from_PIL DONE')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()
if __name__ == '__main__':
unittest.main()