o
    0Îjó  ã                   @   s4   d dl Z d dlZddlmZmZ G dd„ deƒZdS )é    Né   )ÚBaseCVResultÚ	JsonMixinc                   @   s0   e Zd ZdZdejfdd„Zdd„ Zdd„ Zd	S )
ÚDocTrResulta3  
    Result class for DocTr, encapsulating the output of a document image processing task.

    Attributes:
        (inherited from BaseCVResult): Any attributes defined in the base class.

    Methods:
        _to_img(self) -> np.ndarray:
            Converts the stored image result to a numpy array.
    Úreturnc                 C   s   t  | d ¡}d|iS )NÚ	doctr_imgÚres)ÚnpÚarray)ÚselfÚresult© r   úp/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/image_unwarping/result.pyÚ_to_img"   s   zDocTrResult._to_imgc                 O   s4   t  | ¡}| d¡ d|d< tj|g|¢R i |¤ŽS )NÚ	input_imgz...r   )ÚcopyÚdeepcopyÚpopr   Ú_to_str©r   ÚargsÚkwargsÚdatar   r   r   r   &   s   

zDocTrResult._to_strc                 O   s,   t  | ¡}| d¡ tj|g|¢R i |¤ŽS )Nr   )r   r   r   r   Ú_to_jsonr   r   r   r   r   ,   s   

zDocTrResult._to_jsonN)	Ú__name__Ú
__module__Ú__qualname__Ú__doc__r	   Zndarrayr   r   r   r   r   r   r   r      s
    r   )r   Únumpyr	   Zcommon.resultr   r   r   r   r   r   r   Ú<module>   s   