|
|
@@ -60,7 +60,7 @@ class ImageMattingTest(unittest.TestCase): |
|
|
|
cv2.imwrite('result.png', result['output_png']) |
|
|
|
print(f'Output written to {osp.abspath("result.png")}') |
|
|
|
|
|
|
|
@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_modelscope_dataset(self): |
|
|
|
dataset = PyDataset.load('beans', split='train', target='image') |
|
|
|
img_matting = pipeline(Tasks.image_matting, model=self.model_id) |
|
|
|