a
==ico @ 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 g/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/mint/app.py jsonify_nvd3 s r c
C sH zt | }W n4 tyB } z d| t|W Y d }~S d }~0 0 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 mean zm+stdzm-std)nploadtxt Exceptionformatstrospathsplitextbasenamenvd3 lineChartargs chart_heightsampler shapearange add_serier )filenamedatae
chart_namer stepxdatar r r
visualize_summary( s( &
$$r4 c
C s z*t | }|jdkr(|d d t jf }W n4 ty^ } z d| t|W Y d }~S d }~0 0 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 qvt|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/ r0 r1 r r2 r3 Z
trunc_sizeZ running_matir r r
visualize_print_logA sJ
&
r8 c C sL t jtj| }| dr"t|S | dr4t|S tj d
| ddS d S )NsummarylogzUnsupport file: {} r ) r" r# joinr( rootendswithr4 r8 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
zindex..z
index.html)r= namesZdebug_messages) globr" r# r< r( r= sortr Zrender_template)filesrF r r r
index| s rJ z /visualization/c C s t | }|S )N)r@ )r retr r r
visualization s rL 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
ra __main__r ) rR r rG numpyr r& r" sysZtornado.httpserverrW Z tornado.wsgir# abspathdirname__file__Z
__folder__ZFlask__name__r< rX r( r r4 r8 r@ ZrouterJ rL ra r^ r r r r
s4 .
-