B #=6\@sfdZddlZddlmZmZddZddZGdd d eZ e Z Gd d d eZ Gd d d eZ dS)a(Functions that could be used to visualize Tensors. This is adapted from the old-time iceberk package that Yangqing wrote... Oh gold memories. Before decaf and caffe. Why iceberk? Because I was at Berkeley, bears are vegetarian, and iceberg lettuce has layers of leaves. (This joke is so lame.) N)cmpyplotcCs*|j}||d|d|d|dS)zConvert a HWC array to CHW.)ndimswapaxes)arrrr :/tmp/pip-install-l3r2oljg/torch/caffe2/python/visualize.py ChannelFirstsr cCs*|j}||d|d|d|dS)zConvert a CHW array to HWC.rrr)rr)r rr r r ChannelLastsr c@sVeZdZdZdddZdddZddejfdd Zd d Z dejfd d Z ddZ dS)PatchVisualizerz&PatchVisualizer visualizes patches. rcCs ||_dS)N)gap)selfrr r r __init__ szPatchVisualizer.__init__NcCsrt|jdkr |||}n$t|jdkrD|jddkrDtdt|jdkr`|dkr`tj}tj||d|S)zVisualizes one single patch. The input patch could be a vector (in which case we try to infer the shape of the patch), a 2-D matrix, or a 3-D matrix whose 3rd dimension has 3 channels. rrrz$The input patch shape isn't correct.N)cmap) lenshapereshapeget_patch_shape ValueErrorrgrayrimshow)rpatchrr r r ShowSingle#szPatchVisualizer.ShowSinglecCs|jd}|dkr&ttt|}tt|t|}t|jdkrj||jdf||d}t |jdd|j }|t ||g|j }t|jdkr|jddkr||jdd}t |} |dkrt j }n&|jddkrt |d} ntd nt |} |dkr&t j }t| ||} x^t|D]R} | ||d} | ||d} || | | | |jd| | |jdf<qBWtj| |d d td | S) a3Visualize multiple patches. In the passed in patches matrix, each row is a patch, in the shape of either n*n, n*n*1 or n*n*3, either in a flattened format (so patches would be a 2-D array), or a multi-dimensional tensor. We will try our best to figure out automatically the patch size. rNrrr)rz%The input patch shape isn't expected.Znearest)r interpolationoff)rintnpceilsqrtfloatrrrarrayrtuplerrrZonesrangerrZaxis)rpatchesZncolsrbg_funcZ num_patchesZnrowsZpatch_size_expandZ image_sizeZ image_shapeimagepidrowcolr r r ShowMultiple4s:   2 zPatchVisualizer.ShowMultiplecOs<|t|}|t|ttjj}|j|f||S)z~Similar to ShowMultiple, but always normalize the values between 0 and 1 for better visualization of image-type data. )r!minmaxZfinfoZfloat64Zepsr.)rr(argskwargsr r r ShowImages`szPatchVisualizer.ShowImagescCs6t|jdkrtdt|jdd}|j|||dS)z This function shows the channels of a patch. The incoming patch should have shape [w, h, num_channels], and each channel will be visualized as a separate gray patch. rz$The input patch shape isn't correct.rr)rr))rrrr!rTr.)rrrr)Zpatch_reorderedr r r ShowChannelshszPatchVisualizer.ShowChannelscCs^t|j}|t|krJt|jd}|t|kr@td||dfSt|}||fSdS)zGets the shape of a single patch. Basically it tries to interprete the patch as a square, and also check if it is in color (3 channels) g@z#I can't figure out the patch shape.rN)r!r#sizefloorrr )rrZedgeLenr r r rss  zPatchVisualizer.get_patch_shape)r)N) __name__ __module__ __qualname____doc__rrr!Zmeanr.r3r5rr r r r rs  , rc@s<eZdZeddZeddZeddZeddZd S) NHWCcOstj||dS)N)_default_visualizerr)r1r2r r r rszNHWC.ShowSinglecOstj||dS)N)r=r.)r1r2r r r r.szNHWC.ShowMultiplecOstj||dS)N)r=r3)r1r2r r r r3szNHWC.ShowImagescOstj||dS)N)r=r5)r1r2r r r r5szNHWC.ShowChannelsN)r8r9r: staticmethodrr.r3r5r r r r r<s   r<c@s<eZdZeddZeddZeddZeddZd S) NCHWcOstjt|f||dS)N)r=rr )rr1r2r r r rszNCHW.ShowSinglecOstjt|f||dS)N)r=r.r )rr1r2r r r r.szNCHW.ShowMultiplecOstjt|f||dS)N)r=r3r )rr1r2r r r r3szNCHW.ShowImagescOstjt|f||dS)N)r=r5r )rr1r2r r r r5szNCHW.ShowChannelsN)r8r9r:r>rr.r3r5r r r r r?s   r?) r;Znumpyr!Z matplotlibrrr r objectrr=r<r?r r r r  si