Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933]ocr-recognition re-upload

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9826159
This commit is contained in:
yuanzhi.zyz
2022-08-19 19:02:27 +08:00
committed by yingda.chen
parent c6db966a99
commit 4b501dd44a
3 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
data/test/images/ocr_recognition_document.png Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:29f2ad929c852f6456367054d13e113078cf06b763fe54d73fd324f789331aa3
size 61611

5
modelscope/pipelines/builder.py
View File

@@ -124,8 +124,9 @@ DEFAULT_MODEL_FOR_PIPELINE = {
Tasks.image_classification:
(Pipelines.daily_image_classification,
'damo/cv_vit-base_image-classification_Dailylife-labels'),
Tasks.ocr_recognition: (Pipelines.ocr_recognition,
'damo/cv_convnextTiny_ocr-recognition_damo'),
Tasks.ocr_recognition:
(Pipelines.ocr_recognition,
'damo/cv_convnextTiny_ocr-recognition-general_damo'),
Tasks.skin_retouching: (Pipelines.skin_retouching,
'damo/cv_unet_skin-retouching'),
Tasks.crowd_counting: (Pipelines.crowd_counting,

6
tests/pipelines/test_ocr_recognition.py
View File

@@ -19,19 +19,19 @@ from modelscope.utils.test_utils import test_level
class OCRRecognitionTest(unittest.TestCase):
def setUp(self) -> None:
self.model_id = 'damo/cv_convnextTiny_ocr-recognition_damo'
self.model_id = 'damo/cv_convnextTiny_ocr-recognition-general_damo'
self.test_image = 'data/test/images/ocr_recognition.jpg'
def pipeline_inference(self, pipeline: Pipeline, input_location: str):
result = pipeline(input_location)
print('ocr recognition results: ', result)
@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):
ocr_recognition = pipeline(Tasks.ocr_recognition, model=self.model_id)
self.pipeline_inference(ocr_recognition, self.test_image)
@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_from_modelhub_PILinput(self):
ocr_recognition = pipeline(Tasks.ocr_recognition, model=self.model_id)
imagePIL = PIL.Image.open(self.test_image)