a "Df@s"ddlZddlmZd ddZdS) N)resizeFlinearcentercCs|jdd|ddkr<|r8|tj|ddtdfS|St||jkr^t||jdg}tj||jd}|r||ddddf<|jdd\}}d|d|d|d|} } t | | } t t || t t || } } t || | |d}d\}}|jdd\}}|dd\} } |d krX|| kr@| |d}|| kr| |d}nR|d krld}d}n>|d kr|| kr| |}|| kr| |}nt d |||||||||f<|rtj|ddtd}d |||||||f<||fS|SdS)aPCenterize image for specified image size Parameters ---------- src: numpy.ndarray Image to centerize shape: tuple of int Image shape (height, width) or (height, width, channel) cval: int or float or numpy.ndarray Color to be filled in the blank. return_mask: numpy.ndarray Mask for centerized image. interpolation: str Interpolation method (default: 'linear'). loc: str Location of image ('center', 'lt', 'rb'). (default: 'center') Returns ------- dst: numpy.ndarray Centerized image. N)dtypeg?rr)heightwidth interpolation)rrrltrbzUnsupported loc: {}T)shapenpZonesboollenndimlistzerosrminintroundr ValueErrorformat)srcr cvalZ return_maskr locdstZsrc_hZsrc_wZscale_hZscale_wZscaleZdst_hZdst_wphpwhwmaskr"]/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/imgviz/centerize.py centerizesJ" "         r$)NFrr)numpyrrr$r"r"r"r#s