add structured model probing pipeline for image classification

增加对structured model probing pipeline的支持

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11376544
This commit is contained in:
wuzhifan.wzf
2023-02-08 08:29:56 +00:00
committed by wenmeng.zwm
parent 0967ece5a0
commit 7a65cf64e9
10 changed files with 695 additions and 7 deletions

View File

@@ -0,0 +1,29 @@
# Copyright 2021-2022 The Alibaba Fundamental Vision Team Authors. All rights reserved.
import unittest
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.utils.demo_utils import DemoCompatibilityCheck
from modelscope.utils.test_utils import test_level
class ImageStructuredModelProbingTest(unittest.TestCase,
DemoCompatibilityCheck):
def setUp(self) -> None:
self.task = Tasks.image_classification
self.model_id = 'damo/structured_model_probing'
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_run_modelhub(self):
recognition_pipeline = pipeline(self.task, self.model_id)
file_name = 'data/test/images/image_structured_model_probing_test_image.jpg'
result = recognition_pipeline(file_name)
print(f'recognition output: {result}.')
if __name__ == '__main__':
unittest.main()