o
    &jm                     @   sT   d dl Zg dZdd Zdd Zdd Zd	d
 Zdd Zdd Zdd Z	dd Z
dS )    N)visualize_detectionvisualize_keypoint_detectionvisualize_face_detectionvisualize_face_alignmentvisualize_segmentationvisualize_mattingvisualize_ocrvisualize_headposec           
      C      zdd l }W n ty   tdw t|d }t|d }t|d }t|d }|d }|jj }||_||_	||_
||_||_|j| |}	|	S )Nr      fastdeploy is required for visualizing results，please refer to https://github.com/PaddlePaddle/FastDeploy to install fastdeployboxesscores	label_idsmaskscontain_masks)
fastdeploy	ExceptionRuntimeErrornparrayCvisionZDetectionResultr   r   r   r   r   Zvis_detection)
imagedatafdr   r   r   r   r   detection_resultresult r   z/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/visualdl/component/inference/fastdeploy_client/visualizer.pyr      s(   r   c                 C   s~   zdd l }W n ty   tdw t|d }t|d }t|d }|jj }||_||_	||_
|j| |}|S )Nr   r   	keypointsr   
num_joints)r   r   r   r   r   r   r   ZKeyPointDetectionResultr   r   r    Zvis_keypoint_detection)r   r   r   r   r   r    r   r   r   r   r   r   /       r   c                 C   s   zdd l }W n ty   tdw t|d }t|d }t|d }|d }|jj }||_||_	||_
||_|j| |}|S )Nr   r   r   r   	landmarkslandmarks_per_face)r   r   r   r   r   r   r   ZFaceDetectionResultr   r   r"   r#   Zvis_face_detection)r   r   r   r   r"   r#   r   r   r   r   r   r   C   s$   r   c                 C   V   zdd l }W n ty   tdw t|d }|jj }||_|j	| |}|S )Nr   r   r"   )
r   r   r   r   r   r   r   ZFaceAlignmentResultr"   Zvis_face_alignment)r   r   r   r"   Zfacealignment_resultr   r   r   r   r   Y      r   c                 C   s~   zdd l }W n ty   tdw t|d }t|d }t|d }|jj }||_||_	||_
|j| |}|S )Nr   r   r   	score_mapshape)r   r   r   r   r   r   r   ZSegmentationResultr'   r&   r   Zvis_segmentation)r   r   r   r   r&   r'   Zsegmentation_resultr   r   r   r   r   i   r!   r   c           	      C   s   zdd l }W n ty   tdw t|d }t|d }|d }t|d }|jj }||_||_	||_
||_|j| |}|S )Nr   r   alpha
foregroundcontain_foregroundr'   )r   r   r   r   r   r   r   ZMattingResultr(   r)   r*   r'   Zvis_matting)	r   r   r   r(   r)   r*   r'   Zmatting_resultr   r   r   r   r   }   s$   r   c           
      C   r
   )Nr   r   r   text
rec_scores
cls_scores
cls_labels)r   r   r   r   r   r   r   Z	OCRResultr   r+   r,   r-   r.   Z	vis_ppocr)
r   r   r   r   r+   r,   r-   r.   Z
ocr_resultr   r   r   r   r      s(   r   c                 C   r$   )Nr   r   euler_angles)
r   r   r   r   r   r   r   ZHeadPoseResultr/   Zvis_headpose)r   r   r   r/   Zheadpose_resultr   r   r   r   r	      r%   r	   )numpyr   __all__r   r   r   r   r   r   r   r	   r   r   r   r   <module>   s   