o 0j @s@ddlmZmZmZddlZddlmZejGdddZ dS))ListTupleUnionN) benchmarkcseZdZdZddeeedfffdd Zdeee j e e j dffdee j fd d Z d ee j e e j dffde j fd d Z ZS)DocTrPostProcessa^ Post-processing class for cropping regions from images (though currently only performs scaling and color channel adjustments). Attributes: scale (np.float32): A scaling factor to be applied to the image pixel values. Defaults to 255.0 if not provided. Methods: __call__(imgs: List[Union[np.ndarray, Tuple[np.ndarray, ...]]]) -> List[np.ndarray]: Call method to process a list of images. doctr(pred: Union[np.ndarray, Tuple[np.ndarray, ...]]) -> np.ndarray: Method to process a single image or a tuple/list containing an image. Nscalecs8tt|ttfrt||_dStd|_dS)a Initializes the DocTrPostProcess class with a scaling factor. Args: scale (Union[str, float, None]): A scaling factor for the image pixel values. If a string is provided, it will be converted to a float. Defaults to 255.0. go@N)super__init__ isinstancestrfloatnpZfloat32r)selfr __class__t/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/image_unwarping/processors.pyr &s zDocTrPostProcess.__init__imgs.returncsfdd|DS)aW Processes a list of images using the `doctr` method. Args: imgs (List[Union[np.ndarray, Tuple[np.ndarray, ...]]]): A list of images to process. Each image can be a numpy array or a tuple containing a numpy array. Returns: List[np.ndarray]: A list of processed images. csg|]}|qSr)doctr).0Zimgrrr @sz-DocTrPostProcess.__call__..r)rrrrr__call__3s zDocTrPostProcess.__call__predcCs|t|tr |d}n|}t|tjsJd|d}|ddd}||j9}|dddddddf}|jddd }|S) a Processes a single image. Args: pred (Union[np.ndarray, Tuple[np.ndarray, ...]]): The image to process, which can be a numpy array or a tuple containing a numpy array. Only the first element is used if it's a tuple. Returns: np.ndarray: The processed image. Raises: AssertionError: If the input is not a numpy array. rz?Invalid input 'im' in DocTrPostProcess. Expected a numpy array.NZuint8F)copy)r tuplerndarrayZsqueezeZ transposerZastype)rrZimrrrrBs    zDocTrPostProcess.doctr)N)__name__ __module__ __qualname____doc__rr r r rrr!rrr __classcell__rrrrrs  0r) typingrrrnumpyrZutils.benchmarkrZtimeitrrrrrs