U <c@s0ddlZd ddZddZd dd Zd d ZdS)NTcsdddlmddlmmfddt|trTfdd|D}t|S|}|SdS)aARender matplotlib figure to numpy format. Note that this requires the ``matplotlib`` package. Args: figure (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_aggB/tmp/pip-unpacked-wheel-gikjz4vx/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) a! Converts a 5D tensor [batchsize, time(frame), channel(color), height, width] into 4D tensor with dimension [time(frame), new_width, new_height, channel]. 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_power23sz!_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_gridEs$      6 r:cstttks"tdt|jtksFtd|jtdkrfdddD}||}t|}|ddd Std kr؇fd dd D}||}|jddkrt |||gd}|Stdkrfd ddD}||}t |||gd}|SdS)NzLYou can not use the same dimension shordhand twice. input_format: {}z_size of input tensor and input format are different. tensor shape: {}, input_format: {}rcsg|]}|qSrfindrr+ input_formatrrrosz"convert_to_HWC..ZNCHWrrrrcsg|]}|qSrr;r=r>rrrusZHWCcsg|]}|qSrr;r=r>rrr|sZHW) lensetr.formatr!upperr'r:rr%r)Ztensorr?indexZ tensor_NCHWZ tensor_CHWZ tensor_HWCrr>rconvert_to_HWC_s@       rE)T)r-)Znumpyrrr,r:rErrrrs "