o ॵi @sddlmZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZeZe je jejdGd d d eZd S) ) Pipelines)OCRRecognition) OutputKeys)InputPipeline) PIPELINES)Tasks) get_logger) module_namecsJeZdZdZdeffdd ZfddZddZd d Zd d Z Z S) OCRRecognitionPipelineu OCR Recognition Pipeline. Example: ```python >>> from modelscope.pipelines import pipeline >>> ocr_recognition = pipeline('ocr-recognition', 'damo/cv_crnn_ocr-recognition-general_damo') >>> ocr_recognition("http://duguang-labelling.oss-cn-shanghai.aliyuncs.com" "/mass_img_tmp_20220922/ocr_recognition_handwritten.jpg") {'text': '电子元器件提供BOM配单'} ``` modelc s`t|ts Jdtjdd|i|td||j|j|_ |j tddS)z use `model` to create a ocr recognition pipeline for prediction Args: model: model id on modelscope hub or `OCRRecognition` Model. preprocessor: `OCRRecognitionPreprocessor`. zmodel must be a single strr zloading model from dir zloading model doneN) isinstancestrsuper__init__loggerinfor todeviceocr_recognizereval)selfr kwargs __class__r d/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/pipelines/cv/ocr_recognition_pipeline.pyrs  zOCRRecognitionPipeline.__init__c stj|fi|S)a Recognize text sequence in the text image. Args: input (`Image`): The pipeline handles three types of images: - A string containing an HTTP link pointing to an image - A string containing a local path to an image - An image loaded in PIL or opencv directly The pipeline currently supports single image input. Return: A text sequence (string) of the input text image. )r__call__)rinputrrr rr-szOCRRecognitionPipeline.__call__cCs||}|S)N) preprocessorrinputsoutputsr r r preprocess@s z!OCRRecognitionPipeline.preprocesscCs||d}|S)Nimage)rr r r rforwardDzOCRRecognitionPipeline.forwardcCstj|di}|S)Npreds)rTEXTr r r r postprocessHr&z"OCRRecognitionPipeline.postprocess) __name__ __module__ __qualname____doc__rrrr#r%r) __classcell__r r rrr s r N)modelscope.metainfor$modelscope.models.cv.ocr_recognitionrmodelscope.outputsrmodelscope.pipelines.baserrmodelscope.pipelines.builderrmodelscope.utils.constantrmodelscope.utils.loggerr rregister_moduleocr_recognitionr r r r rs