o 0j@sddlZddlZddlmZmZddlmZddlm Z ddl m Z ed r*ddl Z ejGd d d e Zejed Gd d d ZdS)N)class_requires_depsis_dep_available) benchmark)funcs) _BaseResizezopencv-contrib-pythoncs6eZdZdZ d fdd Zd dd Zd d ZZS)ResizezResize the image.FNLINEARcs<tj||dt|tr||f}t|||_||_dS)aC Initialize the instance. Args: target_size (list|tuple|int, optional): Target width and height. -1 will return the images directly without resizing. keep_ratio (bool, optional): Whether to keep the aspect ratio of resized image. Default: False. size_divisor (int|None, optional): Divisor of resized image size. Default: None. interp (str, optional): Interpolation method. Choices are 'NEAREST', 'LINEAR', 'CUBIC', 'AREA', and 'LANCZOS4'. Default: 'LINEAR'. ) size_divisorinterpN)super__init__ isinstanceintFcheck_image_size target_size keep_ratio)selfrrr r __class__z/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/semantic_segmentation/processors.pyr s    zResize.__init__csBdurjnttrftfdd|DS)applyNcsg|]}|qSr)resize).0imgrrrr >sz#Resize.__call__..)rrrrr)rimgsrrr r__call__8s   zResize.__call__cs|dkr|S|jddddd}|ddkr|ddks Jjr6|jdd\}}||f|\}}jrBfdd|D}tj||jd}|S) N)r r rr rcs"g|] }t|jjqSr)mathceilr )rirrrr!Msz!Resize.resize..)r)shaperZ _rescale_sizer rrr)rrrZ original_sizehw_rr(rr@s z Resize.resize)r FNr )N)__name__ __module__ __qualname____doc__rr#r __classcell__rrrrr s  r c@s eZdZdZddZddZdS)SegPostProcesszSemantic Segmentation PostProcess This class is responsible for post-processing detection results, only including restoring the prediction segmentation map to the original image size for now. cs:t|t|ks Jdd|D}fddt||DS)NcSs$g|]}|jdddddqS)Nrr )r))rZ src_imagerrrr!as$z+SegPostProcess.__call__..csg|] \}}||qSr)reverse_resize)rrsrc_sizer(rrr!bs)lenzip)rr"Z src_imagesZ src_sizesrr(rr#^s  zSegPostProcess.__call__cCspt|ttfr t|dksJ|ddkr|ddksJ|jdks$Jtj|d|tjd}tj |dd}|S)aRestore the prediction map to source image size using nearest interpolation. Args: img (np.ndarray): prediction map with shape of (1, width, height) src_size (Tuple[int, int]): source size of the input image, with format of (width, height). rrr$r)Zdsize interpolation)Zaxis) rtuplelistr5ndimcv2rZ INTER_NEARESTnpZ expand_dims)rrr4Z reversed_imgrrrr3fs zSegPostProcess.reverse_resizeN)r-r.r/r0r#r3rrrrr2Us r2)r%numpyr<Z utils.depsrrZutils.benchmarkrZ common.visionrrZcommon.vision.processorsr r;Ztimeitr r2rrrrs   8