U
do @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZ d dl
Z ej ej
eZejeej edej eddZdadd Zdd Zd d
Zd d Zed
dd Zeddd Zdd Z edkre ej dd dS ) NZ templatesZstatic)Ztemplate_folderZ
static_folderc C sB | | jdd }| jd}tj| j| j|| dS )Nzresultscript)Z buildcontentZ htmlcontentfindflaskjsonify containerstrip)chartZ script_startZ
script_end r :/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/mint/app.py jsonify_nvd3 s r c
C sJ zt | }W n6 tk
rD } zd| t| W Y S d }~X Y nX tjtj| d }t j
|d t j dd}t j
dk rt|jd t j
d}nt j
}t d|jd |}|j|||df dd |j|||df d d |j|||d
f d d |j|||d
f ||d f d
d |j|||d
f ||d f dd t|S )NCannot load file {}: {}r Z_summary_chart.03gnameheightZ
y_axis_format minxyr max Zmean zm+stdzm-std)nploadtxt Exceptionformatstrospathsplitextbasenamenvd3 lineChartargs chart_heightsampler shapearange add_serier )filenamedatae
chart_namer stepxdatar r r
visualize_summary( s( &
$$r3 c
C s z*t | }|jdkr(|d d t jf }W n6 tk
r` } zd| t| W Y S d }~X Y nX tj tj
| d }t j |d t
jdd}t
jdk rt|jd t
j d}nt
j}t d|jd |}|jd dkr`|jd | }|d || ||f}|j|d | |jdddd |j|d | |jddd
d |j|||df |d n@tdt|jd t
jD ]&} |j|||| f d || d qxt|S ) Nr r r Z
_log_chartr r )ZaxisZ running_minr Z running_maxz{}[{}])r r ndimZnewaxisr r r r! r" r# r$ r% r& r' r( r) r r* r+ Zreshaper, r rangeZ
max_curvesr )
r- r. r/ r0 r r1 r2 Z
trunc_sizeZ running_matir r r
visualize_print_logA sJ
&
r7 c C sL t jtj| }| dr"t|S | dr4t|S tj d
| ddS d S )NsummarylogzUnsupport file: {} r ) r! r" joinr' rootendswithr3 r7 r r r )r- fullnamer r r
visualize_fileo s
r? /c C s@ t tjtjd} | dd | D }tjdtj||dS )Nz*.*c S s g | ]}t j|qS r )r! r" r$ ).0fr r r
s zindex..z
index.html)r< namesZdebug_messages) globr! r" r; r' r< sortr Zrender_template)filesrD r r r
index| s rH z /visualization/c C s t | }|S )N)r? )r retr r r
visualization s rJ c C s t d}|jddtddd |jddtd d
d |jd td d
d |jdtddd |jddtddd || atj tj
t }|
tj tdtj tjj d S )NzThe mint visualizer.z-pz--porti zThe flask port to use.)typedefaulthelpz-rz--root.z0The root folder to read files for visualization.z --max_curves z2The max number of curves to show in a dump tensor.z--chart_heighti, zThe chart height for nvd3.z-sz--samplei8zSample every given number of data points. A negative number means the total points we will sample on the whole curve. Default 100 points.z#Tornado server starting on port {}.)argparseArgumentParser add_argumentintr
parse_argsr' tornadoZ
httpserverZ
HTTPServerZwsgiZ
WSGIContainerapplistenportprintr ZioloopZIOLoopinstancestart)argvparserserverr r r
main sN
r_ __main__r ) rP r rE Znumpyr r% r! sysZtornado.httpserverrU Z tornado.wsgir" abspathdirname__file__Z
__folder__ZFlask__name__r; rV r' r r3 r7 r? ZrouterH rJ r_ r\ r r r r
s4 .
-