Vc@@sddlmZddlZddlZyddlmZWn!ek r_ddlmZnXddlZ ddl Z ddl Z dZd Zd Zddd Zd Zeed ddZdeddZdZdS(i(tabsolute_importN(tStringIOs.pngs.jpgs.jpegs.bmps.ppmcC@sjd}tjj|rdy tjj|}|jWqttk r`}t j d|j qtXnt j d||j dkr|S|j dkr|j dS|j dkrtjjd|jd}|j|d|j d |S|j dkr |j dS|j dkrStjjd|jd}|j|d|j d |St j d |j dS(s Reads a file from `path` and returns a PIL.Image with mode 'L' or 'RGB' Raises LoadImageError Arguments: path -- path to the image, can be a filesystem path or a URL s IOError: %ss"%s" not foundtLtRGBt1tLAitmasktRGBAtCMYKtYCbCrtPsImage mode "%s" not supportedN(RsRGB(R(sLA(sCMYKsYCbCr(R sRGBA(iii(tNonetostpathtexiststPILtImagetopentloadtIOErrorterrorstLoadImageErrortmessagetmodetconverttnewtsizetpaste(R timageteR((s9/home/mohanty/data/final_dataset/utils/image_processor.pyt load_image$s.  cC@s t|tjs!tdn|dkr@td|nttj|jd|}ttj|jd|}|jd}tj|||fdtj}xWtj ||fD]@\}}|tj||tj||f|||f1) sExpected ndarrayis'Ratio must be greater than 1 (ratio=%f)iitdtype( t isinstancetnptndarrayt ValueErrortinttmathtfloortshapetuint8tndindex(Rtratiotwidththeighttchannelstouttxty((s9/home/mohanty/data/final_dataset/utils/image_processor.pytupscaleLs    !"8cC@s|dkrd}n|dkr4td|n|dkrStd|nt|tjjr|dkr|j}|d krd}q|d krd}qtd |n*|dkrd }n|dkrd }n|j|kr|j|}ntj|}nt|tj r|j tj krP|j tj }n|j dkr|jd dkr|j|jd }n|dkr|j d krd}q|j dkr|jd dkrd}qtd |jfq|dkrv|j d kr|j dkrZ|jd dkrZtj|dddgj tj }qstd |jfqq|dkr|j d krtj|dj|jd}q|jd dkrtd |jfqqn td|jd|kr|jd|kr|Sd}t|jd|}t|jd|}|dksj||krtjj|||fd|S|dkr||kr|} tt|jd|} n#|} tt|jd|} tjj|| | fd|}||krOtt| |d} |dd| | |fStt| |d} || | |ddfSn)|dkrQ||kr|} tt|jd|} || d dkr-| d7} q-nD|} tt|jd|} || d dkr-| d7} ntjj|| | fd|}ne|dkr||d} tt|jd| } tt|jd| } ||kr|| d dkr| d7} n-||kr|| d dkr| d7} ntjj|| | fd|}||kritt| |d} |dd| | |f}qtt| |d} || | |ddf}ntd|||kr:|| d } | |f}|dkr||f7}ntjjdd|j d}tj|||fdd}nu|| d } || f}|dkrp||f7}ntjjdd|j d}tj|||fdd}|SdS(sN Resizes an image and returns it as a np.array Arguments: image -- a PIL.Image or numpy.ndarray height -- height of new image width -- width of new image Keyword Arguments: channels -- channels of new image (stays unchanged if not specified) resize_mode -- can be crop, squash, fill or half_crop tsquashtcroptfillt half_cropsresize_mode "%s" not supportediis"unsupported number of channels: %sRRsunknown image mode "%s"isinvalid image shape: %sgA`"?gbX9?gv/?s<resize_image() expected a PIL.Image.Image or a numpy.ndarrayitbilineartinterpg@Nsunrecognized resize_mode "%s"iR(taxis(scropssquashsfills half_crop(Nii(i(R R#R RRRRR!tarrayR"RR(tastypetndimR'treshapetdottrepeattfloattscipytmisctimresizeR$troundt Exceptiontrandomtrandintt concatenate(RR,R+R-t resize_modet image_modeR7t width_ratiot height_ratiot resize_heightt resize_widthtstartt new_ratiotpaddingt noise_sizetnoise((s9/home/mohanty/data/final_dataset/utils/image_processor.pyt resize_image`s              "  "  "* % &   ! !   $     ! ##   !!  !cC@s|dkrdSt|tjjr(n3t|tjrOtjj|}n td|j}|ssd}n |j }t }|j |d||j j djdd}d||fS( s Returns an image embedded in HTML base64 format (Based on Caffe's web_demo) Arguments: image -- a PIL.Image or np.ndarray s)image must be a PIL.Image or a np.ndarraytjpegtformattbase64s tsdata:image/%s;base64,%sN(R R RRR!R"t fromarrayR#RUtlowerRtsavetgetvaluetencodetreplace(Rtfmtt string_buftdata((s9/home/mohanty/data/final_dataset/utils/image_processor.pytembed_image_htmls      !idic C@s|jdkr1|ddtjtjf}n|jdkrm|j|jd|jdddf}n|jdkr|jddkr|dddgdf}|jddd}|tjdf}qn|jdkr|jddkrD|jdddd}|dddddddddgf}q|jddkr|jdddd}|dddddddddgf}q|j|jd|jd|jd|jdf}ntd|jfd}|jd}||kr;|ddd|d|f}nF|dkrZd}|d7}nt||d}||9}|| }| r|jdkr|dtjf}nt|d |d |}t |jd d} | |kr |t | } t || }n|S( s2 Returns a vis_square for the given layer data Arguments: data -- a np.ndarray Keyword arguments: allow_heatmap -- if True, convert single channel images to heatmaps normalize -- whether to normalize the data when visualizing max_width -- maximum width for the vis_square iNiii.isunrecognized data shape: %stpadsizet normalize( R;R!tnewaxisR<R't transposet RuntimeErrortmaxt vis_squaretsortedR?R1( R`t allow_heatmapRct min_img_dimt max_widthRbR+tntvist_minR*((s9/home/mohanty/data/final_dataset/utils/image_processor.pytget_layer_vis_squaresN"-44;  (       itjetcC@sd|jkodkns+td|jd}|r||j8}|jdkr||j:}|d9}qn|jdkrt|\}}}tj|t|ddt t||}tj|t|ddt t||}tj|t|ddt t||} tj |d tj f|d tj f| d tj ffd d}tj |d}tj |d}n|jd }ttjtj|jd} | } | } |jd} | | d| kr| d} nd| | | fd|fd|ffd|jd}tj||d d dd}|j| | f|jdjdttd|jd}|j| |jd| |jdf|jd}|jddkrtj|jgd}n|S(s Visualize each image in a grid of size approx sqrt(n) by sqrt(n) Returns a np.array image (Based on Caffe's filter_visualization notebook) Arguments: images -- an array of shape (N, H, W) or (N, H, W, C) if C is not set, a heatmap is computed for the result Keyword arguments: padsize -- how many pixels go inbetween the tiles normalize -- if true, scales (min, max) across all images out to (0, 1) colormap -- a string representing one of the supported colormaps iisimages.ndim must be 3 or 4tfloat32iiigo@.R8R(Rtconstanttconstant_valuesii(ii((ii(iiii(R;tAssertionErrorR:tminRgt get_color_mapR!R7tlentxrangeRGRdtminimumtmaximumR$tceiltsqrtR'tpadR<Rettupletrangetdstacktsqueeze(timagesRbRctcolormaptredmaptgreenmaptbluemaptredtgreentblueRmtnytnxtlengthRPtpaddedttiles((s9/home/mohanty/data/final_dataset/utils/image_processor.pyRhhs:+333E%  8C6c@C@s+dg}dg}dg}|dkrNddg}ddg}ddg}n|dkrddddg}ddddg}ddddg}nd|dkrddddd d d d d dddddddddddddddddddddddddddddddddddddddddddddddddddddddg@}dddddddddddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7dddddddddddddg@}dddddddddddddddddddddddddddddddddddddddddddddddddddd8d9d:d;d<d=d>d?d@dAdBdCdg@}n|dDkrUdddddddddddddddddddddddddddEdFdGdHdIdJdKdLdMdNdOdPddddddddddddddQdRdSdTdUdVdWdXdYd+dZd[d\g@}ddddd d d d d dddddddddddddddddddddddddddddd]d5d^d_d\d`d+dadWd%dTdbdQdddddddddddddg@}dddddddddddddddddddddddddddddddddddddddcdddedfdgdhdidjdkdldmdndodddddddddddddg@}n|dpkr|ddg}ddqg}n{|drkrds|GHndddddqddddqg }dddqddddqddg }dqddddqddddg }dttj|dttj|dttj|fS(us Return a colormap as (redmap, greenmap, bluemap) Arguments: name -- the name of the colormap. If unrecognized, will default to 'jet'. itwhiteitsimplethotgEQEQ?gEQEQ?gyy?gEQEQ?gYeYe?gyy?grq?gEQEQ?gm۶m?gYeYe?g?gyy?g  ?grq?g0 0 ?gEQEQ?geYeY?gm۶m?g  ?gYeYe?g?g?gM4M4?gyy?g?gAA?g$I$I?g#qq?g4M4M?gaa?gXu]u?gQEQE?g<s.       (   P ?