
    Wpf
                         d Z ddlZddlmZ ddlmZ ddlZddlZddlm	Z
 dZ G d dej                  Zed	k    r ej                     dS dS )
z9Tests for mediapipe.python.solutions.selfie_segmentation.    N)absltest)parameterized)selfie_segmentationz#mediapipe/python/solutions/testdatac                   r    e Zd Zdej        dej        fdZd Zd Z ej	        dd          d             Z
d	S )
SelfieSegmentationTestframemaskc                 T   t          j        |t          j        |fdz  d                    }t          j                            t                                          |                                 	                    d          d         dz             }t          j        ||           d S )N   )axis.z.png)npminimumstackospathjointempfile
gettempdiridsplitcv2imwrite)selfr   r	   r   s       s/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/mediapipe/python/solutions/selfie_segmentation_test.py_drawzSelfieSegmentationTest._draw    s    Jubhw{<<<==E7<<++--twwyys/C/CB/G&/PQQDKe    c           	      T   t          j                    5 }|                     t          d          5  |                    t          j        dt
          j                                      ddd                     d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nz0Input image must contain three channel rgb data.$   dtyper      )	mp_selfie_segmentationSelfieSegmentationassertRaisesRegex
ValueErrorprocessr   arangeuint8reshape)r   r   s     r   test_invalid_image_shapez/SelfieSegmentationTest.test_invalid_image_shape%   s$   		2	4	4 <8K!!
HJ J < <##Ib)))11!Q::	< 	< 	<< < < < < < < < < < < < < < << < < < < < < < < < < < < < < < < <s5   BA	B9BB			BB		BB!$B!c                    t          j                    5 }t          j        g dt          j                  }|                    d           |                    |          }|j        dz                      t                    }| 
                    t          j        |          d           d d d            d S # 1 swxY w Y   d S )N)d   r.   r   r!         )r$   r%   r   zerosr*   fillr(   segmentation_maskastypeint
assertLessamax)r   r   imageresultsnormalized_segmentation_masks        r   test_blank_imagez'SelfieSegmentationTest.test_blank_image,   s    		2	4	4 @8Kh}}}BH555ejjooo#++E22g&-&?&)'*+16#;; #
oobg:;;Q???@ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s   BB88B<?B<)generalr   )	landscaper0   c                    t           j                            t           j                            t                    d          }t          j        |          }t          j        |          5 }|	                    t          j
        |t
          j                            }|j        dz                      t                    }|                     |                                |           d d d            d S # 1 swxY w Y   d S )Nztestdata/portrait.jpg)model_selectionr/   )r   r   r   dirname__file__r   imreadr$   r%   r(   cvtColorCOLOR_BGR2RGBr3   r4   r5   r   copy)r   r?   
image_pathr8   r   r9   r:   s          r   test_segmentationz(SelfieSegmentationTest.test_segmentation5   s   bgooh7757 7JJz""E		2'
) 
) 
) =,?#++
,uc/
0
02 2g&-&?&)'*+16#;; #
jj;<<<= = = = = = = = = = = = = = = = = =s   ,A=C66C:=C:N)__name__
__module____qualname__r   ndarrayr   r,   r;   r   named_parametersrG    r   r   r   r      s         2:    
< < <@ @ @ "=!.2BCC
= 
= DC
= 
= 
=r   r   __main__)__doc__r   absl.testingr   r   r   numpyr   mediapipe.python.solutionsr   r$   TEST_IMAGE_PATHTestCaser   rH   mainrM   r   r   <module>rV      s    @ ? 				 ! ! ! ! ! ! & & & & & & 



     U T T T T T7"= "= "= "= "=]3 "= "= "=J z(-///// r   