3
Eg @ sf d Z ddlZddlmZmZ dd Zdd ZG dd d eZ e Z
G d
d d eZ G d 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)cmpyplotc C s* | j }| j|d |d j|d |d S )zConvert a HWC array to CHW. )ndimswapaxes)arrr r
C/usr/local/lib64/python3.6/site-packages/caffe2/python/visualize.py ChannelFirst s r c C s* | j }| j|d |d j|d |d S )zConvert a CHW array to HWC.r r r )r r )r r r
r
r ChannelLast s r
c @ sV e Zd ZdZdddZdddZddejfdd Zd
d Z dejfd d
Z
dd Z dS )PatchVisualizerz&PatchVisualizer visualizes patches.
r c C s
|| _ d S )N)gap)selfr r
r
r __init__ s zPatchVisualizer.__init__Nc C sr t |jdkr |j| j|}n$t |jdkrD|jd dkrDtdt |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.
r r r z$The input patch shape isn't correct.N)cmap) lenshapereshapeget_patch_shape
ValueErrorr grayr imshow)r patchr r
r
r
ShowSingle# s zPatchVisualizer.ShowSinglec C s |j d }|dkr&ttjtj|}ttj|t| }t|j dkrj|j|j d f| j|d }tj |j dd | j
}|tj ||g | j
}t|j dkr|j d dkr|j|j dd }t |} |dkrt j
}n&|j d dkrt |d } ntdnt |} |dkr&t j
}tj| || }
x^t|D ]R} | | |d } | | |d }
|| |
| | |j d |
|
|j d f< qBW tj|
|dd tjd
|
S )
a3 Visualize 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.
r Nr r r z%The input patch shape isn't expected.Znearest)r
interpolationoff)r )r intnpceilsqrtfloatr r r arrayr tupler r r Zonesranger r Zaxis)r patchesZncolsr bg_funcZ num_patchesZnrowsZpatch_size_expandZ
image_sizeZ image_shapeimagepidrowcolr
r
r ShowMultiple4 s:
2
z PatchVisualizer.ShowMultiplec O s< |t j| }|t j|t jt jj }| 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. )r r( argskwargsr
r
r
ShowImages` s zPatchVisualizer.ShowImagesc C s6 t |jdkrtdtj|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.
r z$The input patch shape isn't correct.r r )r r) )r r r r! r Tr. )r r r r) Zpatch_reorderedr
r
r ShowChannelsh s z PatchVisualizer.ShowChannelsc C s^ t j|j}|t j|krJt j|jd }|t j|kr@td||dfS t|}||fS dS )zGets the shape of a single patch.
Basically it tries to interpret 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.r N)r! r# sizefloorr r )r r ZedgeLenr
r
r r s s
z PatchVisualizer.get_patch_shape)r )N) __name__
__module__ __qualname____doc__r r r! Zmeanr. r3 r5 r r
r
r
r r s
, r c @ s< e Zd Zedd Zedd Zedd Zedd Zd S )
NHWCc O s t j| | d S )N)_default_visualizerr )r1 r2 r
r
r r s zNHWC.ShowSinglec O s t j| | d S )N)r= r. )r1 r2 r
r
r r. s zNHWC.ShowMultiplec O s t j| | d S )N)r= r3 )r1 r2 r
r
r r3 s zNHWC.ShowImagesc O s t j| | d S )N)r= r5 )r1 r2 r
r
r r5 s zNHWC.ShowChannelsN)r8 r9 r: staticmethodr r. r3 r5 r
r
r
r r< s r< c @ s< e Zd Zedd Zedd Zedd Zedd Zd S )
NCHWc O s t jt| f|| d S )N)r= r r
)r r1 r2 r
r
r r s zNCHW.ShowSinglec O s t jt| f|| d S )N)r= r. r
)r r1 r2 r
r
r r. s zNCHW.ShowMultiplec O s t jt| f|| d S )N)r= r3 r
)r r1 r2 r
r
r r3 s zNCHW.ShowImagesc O s t jt| f|| d S )N)r= r5 r
)r r1 r2 r
r
r r5 s zNCHW.ShowChannelsN)r8 r9 r: r> r r. r3 r5 r
r
r
r r? s r? )
r; Znumpyr! Z
matplotlibr r r r
objectr r= r< r? r
r
r
r