o ߥiW @sddlZddlZddlmZmZddlZddlZddlZddl Z ddl m Z ddl m Z mZddlmZddlmZddlmZmZmZejeje jdGd d d e ZdS) N)AnyDict) Preprocessors) Preprocessor load_image) PREPROCESSORS)Config)FieldsModeKeys ModelFile) module_namecs4eZdZejfdedeffdd ZddZZS)OCRDetectionPreprocessor model_dirmodecs2t|ttj|tj}|j j j |_ dS)zThe base constructor for all ocr recognition preprocessors. Args: model_dir (str): model directory to initialize some resource mode: The mode for the preprocessor. N) super__init__r from_fileospathjoinr CONFIGURATIONmodelinference_kwargsimage_short_side)selfrrcfgs __class__c/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/cv/ocr_detection/preprocessor.pyrs z!OCRDetectionPreprocessor.__init__c Cs2t|tr tt|}nt|tjjrt|}nt|tjr#|}n tdt ||dddddddf}|j \}}}||krW|j }t t |||dd}n|j }t t |||dd}t|||f}|tjgdtjd}|d}t|dd d d }|||gd } | S) arprocess the raw input data Args: inputs: - A string containing an HTTP link pointing to an image - A string containing a local path to an image - An image loaded in PIL(PIL.Image.Image) or opencv(np.ndarray) directly, 3 channels RGB Returns: outputs: the preprocessed image z:inputs should be either str, PIL.Image, np.array, but got N )gQ^@gR1]@g\(Y@)dtypego@r)img org_shape) isinstancestrnparrayrPILImagendarray TypeErrortypeshaperintmathceilcv2resizefloat32torch from_numpypermutefloat unsqueeze) rinputsr%heightwidth_ new_height new_width resized_imgresultrrr__call__"s8         z!OCRDetectionPreprocessor.__call__) __name__ __module__ __qualname__r INFERENCEr(rrD __classcell__rrrrr s r )r2rtypingrrr4numpyr)r+r7modelscope.metainformodelscope.preprocessorsrr modelscope.preprocessors.builderrmodelscope.utils.configrmodelscope.utils.constantr r r register_modulecv ocr_detectionr rrrrs