o *jW @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 Z CONFIGURATIONmodelZinference_kwargsimage_short_side)selfrrZcfgs __class__p/var/www/html/Deteccion_Ine/venv/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@)Zdtypego@r)imgZ org_shape) isinstancestrnparrayrPILZImageZndarray TypeErrortypeshaperintmathceilcv2resizeZfloat32torchZ from_numpyZpermutefloatZ unsqueeze) rZinputsr!heightwidth_Z new_heightZ new_widthZ resized_imgresultrrr__call__"s8         z!OCRDetectionPreprocessor.__call__) __name__ __module__ __qualname__r Z INFERENCEr#rr5 __classcell__rrrrr s r )r+rtypingrrr-numpyr$r&r/Zmodelscope.metainforZmodelscope.preprocessorsrrZ modelscope.preprocessors.builderrZmodelscope.utils.configrZmodelscope.utils.constantr r r Zregister_moduleZcvZ ocr_detectionr rrrrs