o vZh[@s<ddlZddlmZd ddZddZd dd Zd d ZdS)NTcs`ddlmddlmmfddt|tr*fdd|D}t|S|}|S)aBRender matplotlib figure to numpy format. Note that this requires the ``matplotlib`` package. Args: figures (matplotlib.pyplot.figure or list of figures): figure or a list of figures close (bool): Flag to automatically close the figure Returns: numpy.array: image in [CHW] order rNcs||}|tj|tjd}|j\}}|||dgddddddf}tj |ddd}r< ||S)Ndtyper)source destination) ZFigureCanvasAggZdrawnpZ frombufferZ buffer_rgbauint8canvasZget_width_heightreshapeZmoveaxisclose)figurer datawhZ image_hwcZ image_chw)r pltplt_backend_aggM/var/www/auris/lib/python3.10/site-packages/torch/utils/tensorboard/_utils.py render_to_rgbs & z&figure_to_image..render_to_rgbcsg|]}|qSrr).0r)rrr "sz#figure_to_image..) Zmatplotlib.pyplotZpyplotZmatplotlib.backends.backend_aggbackendsZ backend_agg isinstancelistr stack)Zfiguresr Zimagesimager)r rrrrfigure_to_images   rc Cs|j\}}}}}|jtjkrt|d}dd}||jdsBtd|jd|jd}tj|tj|||||fdfdd}d|dd}|jd|} tj ||| ||||fd }tj |d d }tj ||||| ||fd }|S) aL Convert a 5D tensor into 4D tensor. Convesrion is done from [batchsize, time(frame), channel(color), height, width] (5D tensor) to [time(frame), new_width, new_height, channel] (4D tensor). A batch of images are spreaded to a grid, which forms a frame. e.g. Video with batchsize 16 will have a 4x4 grid. go@cSs|dko ||d@dkS)Nrr)numrrr is_power28sz!_prepare_video..is_power2rr)shape)Zaxisr)Znewshape)rrrrr)Zaxes) r"rr r Zfloat32int bit_length concatenatezerosr transpose) Vbtcrrr!Z len_additionZn_rowsZn_colsrrr_prepare_video)s  $r-c Cst|tjs Jd|jddkrt|||gd}|jdkr&|jddks(J|jd}|jd}|jd}t||}ttt ||}tj d||||f|j d}d}t |D].}t |D]'} ||krjn |||dd|||d|| || d|f<|d}qbq\|S)Nz*plugin error, should pass numpy array hererrrrrr) rr Zndarrayr"r&ndimminr$ceilfloatr'rrange) IZncolsZnimgHWZnrowsr iyxrrr make_gridJs&      6 r:cstttksJdt|jtks%Jd|jdtdkrHfdddD}||}t|}|dd d Std krnfd dd D}||}|jd dkrlt|||gd }|Std krfdddD}||}t|||gd }|SdS)NzJYou can not use the same dimension shordhand twice. input_format: zKsize of input tensor and input format are different. tensor shape: z, input_format: rcg|]}|qSrfindrr, input_formatrrroz"convert_to_HWC..ZNCHWrrrrcr;rr<r>r?rrrurAZHWCcr;rr<r>r?rrr|rAZHW) lensetr"upperr(r:r r&r)Ztensorr@indexZ tensor_NCHWZ tensor_CHWZ tensor_HWCrr?rconvert_to_HWCds@        rF)T)r.) numpyr Z numpy.typingtypingZnptrr-r:rFrrrrs   " !