U Mh?@s0ddlZd ddZddZd dd Zd d ZdS)NTcsdddlmddlmmfddt|trTfdd|D}t|S|}|SdS)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}rx ||S)Ndtyper)sourceZ destination) ZFigureCanvasAggZdrawnpZ frombufferZ buffer_rgbauint8canvasZget_width_heightreshapeZmoveaxisclose)figurer datawhZ image_hwcZ image_chw)r pltplt_backend_aggP/var/www/html/venv/lib/python3.8/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 isinstancelistrstack)Zfiguresr Zimagesimager)r rrrrfigure_to_images   rc Cs|j\}}}}}|jtjkr*t|d}dd}||jdstd|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_power27sz!_prepare_video..is_power2rr)shape)Zaxisr)Znewshape)rrrrr)Zaxes) r!rrr Zfloat32int bit_length concatenatezerosr transpose) Vbtcrrr Z len_additionZn_rowsZn_colsrrr_prepare_video(s   $r,c Cst|tjstd|jddkr4t|||gd}|jdkrL|jddksPt|jd}|jd}|jd}t||}tt t ||}tj d||||f|j d}d}t |D]\}t |D]N} ||krq|||dd|||d|| || d|f<|d}qq|S)Nz*plugin error, should pass numpy array hererrrrrr)rrZndarrayAssertionErrorr!r%ndimminr#ceilfloatr&rrange) IZncolsZnimgHWZnrowsr iyxrrr make_gridIs$      6 r:cs tttks"tdt|jtksJtd|jdtdkrfdddD}||}t|}|dd d Std kr܇fd dd D}||}|jd dkrt|||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: rcsg|]}|qSrfindrr+ input_formatrrrnsz"convert_to_HWC..ZNCHWrrrrcsg|]}|qSrr;r=r>rrrtsZHWCcsg|]}|qSrr;r=r>rrr{sZHW) lensetr.r!upperr'r:rr%r)Ztensorr?indexZ tensor_NCHWZ tensor_CHWZ tensor_HWCrr>rconvert_to_HWCcs>       rD)T)r-)numpyrrr,r:rDrrrrs "!