BVhA d Z ddlZddlZddlZddlZddlZej j e d j ej j dd sddl m
Z nddlm
Z dZdZd Zd Zd
Z G d d Z G d
d Z G d d Zd ZddZd Zd Zd Z ddZ d Z e dk( r e ej@ yy)z}This tool creates an html visualization of a TensorFlow Lite graph.
Example usage:
python visualize.py foo.tflite foo.html
Ntflite_runtime visualize)schema_py_generateda
a
c v t j j j D ] \ }}|| k( s |c S y)z4Converts a numerical enum to a readable tensor type.N) schema_fb
TensorType__dict__items) tensor_typenamevalues O/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/lite/tools/visualize.pyTensorTypeToNamer s< ))2288: kdE
k
c v t j j j D ] \ }}|| k( s |c S y)z3Converts a builtin op code enum to a readable name.N)r BuiltinOperatorr r
)coder r
s r BuiltinCodeToNamer s; ..77==? kdE }
k
r c n t | t r| S d}| | D ] }|t t | z } |S )z;Converts a list of integers to the equivalent ASCII string. )
isinstancestrchrint) name_listresultvals r NameListToStringr sC 3
F (##c#h-'( Mr c e Zd ZdZd Zd Zy) OpCodeMapperz#Maps an opcode index to an op name.c i | _ t |d D ]N \ }}t |d | j |< | j | dk( s4t |d | j |< P y )Noperator_codes builtin_codeCUSTOM custom_code) code_to_name enumerater r )selfdataidxds r __init__zOpCodeMapper.__init__ sq DD!1 23 DQ 0>1B Cd
3 8 +!1!M2B!C# Dr c P || j vrd}n| j | }d||fz S )Nz z%s (%d))r& )r( xss r __call__zOpCodeMapper.__call__ s4 !!!
a
A
a 1v r N__name__
__module__ __qualname____doc__r, r0 r r r r s +D r r c e Zd ZdZd Zy)DataSizeMapperz(For buffers, report the number of bytes.c $ |dt | z S y)Nz%d bytesz--)len)r( r. s r r0 zDataSizeMapper.__call__ s }
#a&
r N)r2 r3 r4 r5 r0 r6 r r r8 r8 s
0r r8 c e Zd ZdZd Zd Zy) TensorMapperzGMaps a list of tensor indices to a tooltip hoverable indicator of more.c || _ y N)r) )r(
subgraph_datas r r, zTensorMapper.__init__ s DIr c R d}||S |dz
}|D ]~ }| j d | }|t | dz z
}|t |d dz z
}|t |d dz z
}|d|v rt |d ndz
}|d |v rt |d ndd
z z
} |d z
}|t | z
}|d z
}|S ) Nr z3tensors r typeshapez[]shape_signaturez z)r) r r r repr)r( r. htmlitensors r r0 zTensorMapper.__call__ s
Dy
k AAD
Byy#A&f
c!fsld
vf~.44d
vf~.44d
6(9tF7O$tDd
#v- F,-.376B Bd
B IDDGODID Kr Nr1 r6 r r r< r< s O r r< c
d }d }g }g }i }i }d} d}
t |d xs g D ] \ } } | d Pt | d D ]? \ }
}||vr| dz
dz | z |
dz |
z f||< |j || || d A | d
Pt | d
D ]? \ }}||vr| dz dz | z |dz |
z f||< |j || || d A |j || || d d
| | dz | z d t |d D ]L \ }}||v r|| n
||v r|| nd}|j || dt |dg |fz d|d |d d N t j ||d }t
|| fz }|S )zAProduces the HTML required to have a d3 visualization of the dag.c d| z S )Nzt%dr6 r* s r
TensorNamez!GenerateGraph..TensorName" 3;r c d| z S )Nzo%dr6 rL s r OpNamez GenerateGraph..OpName% rN r operatorsinputsg ? )sourcetargetoutputs)rW rV opcode_index )idr groupr. yrA )r r z%r (%d)rD r )nodesedges)r' appendgetattrjsondumps_D3_HTML_TEMPLATE) subgraph_idxg
opcode_mapperrM rP r_ r^ firstsecond
pixel_mult
width_multop_indexoptensor_input_position tensor_indextensor_output_positionrI initial_y graph_strrG s r
GenerateGraphrs s0
%
%
%
&** + 4"5 lh (| 1:2h<1H
-
u $"*S.1"4
!B"7!";z!I!K%
. X&
)} 2;ByM2J
.
, v %#+c>A#5"C#9A#="K"M&
. X&
LLX b 01
lj
( )6 () 5 lF +u 4l , 6| F
LL &WVWb 9< HH
q\
q\ jj559:) i 6 6$
+r c d}|dz
}|dz
}|r|dz
}|D ]
\ }}|d|z z
} |dz
}t | D ]B \ }}|dz
}|r|d|z z
}|D ]$ \ }}||v r|| nd}||n || }|d |z z
}& |dz
}D |d
z
}|S ) a Given a list of object values and keys to print, make an HTML table.
Args:
items: Items to print an array of dicts.
keys_to_print: (key, display_fn). `key` is a key in the object. i.e.
items[0][key] should exist. display_fn is the mapping function on display.
i.e. the displayed html cell will have the string returned by
`mapping_fn(items[0][key])`.
display_index: add a column which is the index of each row in `items`.
Returns:
An html table.
r z
z
z
index
z
%s
z
z
%d
Nz
%s
)r' ) r
keys_to_print
display_indexrG hmapperr* rI r s r GenerateTableHtmlrz [ s
$/$($ D iaMA D )$ u%
kc6HD
mc!!d" $ 6fF1I$c>Cvc{c
o##d$
ID
,$
+r c z t j dd| }t j dd| j S )z2Converts an identifier in CamelCase to snake_case.z(.)([A-Z][a-z]+)z\1_\2z([a-z0-9])([A-Z]))resublower)camel_case_inputs1s r CamelCaseToSnakeCaser s3 vv (,<=" #Xr 2 8 8 ::r c t | t s t | t st | t r| S t | d r]i }t | D ]K }| j
| }t | r |d dk7 s)t | }|dk( rdn|}t || ||< M |S t | t j r|r| S | j S t | d r| D cg c] }t || c}S | S c c}w )a Converts a hierarchy of FB objects into a nested dict.
We avoid transforming big parts of the flat buffer into python arrays. This
speeds conversion from ten minutes to a few seconds on big graphs.
Args:
fb: a flat buffer structure. (i.e. ModelT)
preserve_as_numpy: true if all downstream np.arrays should be preserved.
false if all downstream np.array should become python arrays
Returns:
A dictionary representing the flatbuffer rather than a flatbuffer object.
r r _buffersT__len__)
r r floatr hasattrdir__getattribute__callabler FlatbufferToDictnpndarraytolist)fbpreserve_as_numpyr attribute_name attribute
snake_namepreserveentrys r r r s CJr5 1ZC5H
Ir:
F b' C%%n5i
i
^A%6#%=).9
)Y64TensorFlow Lite ModelzNull (used model object)filename))r NversionN) descriptionNz
c | S r> r6 )r. s r z create_html.. s ! r z
z
z
ru r) r# r% r r" deprecated_builtin_code subgraphsz
rT rX )builtin_optionsNrY r rC )rD N)rE N)bufferN) quantizationNz
Subgraph %d
z
Inputs/Outputs
)rT rX F)rw z
Tensors
rA rS z
Ops
z6
z
z
Buffers
r z
Operator Codes
z
)ospathexists RuntimeErrorendswithopen bytearrayreadr rb load_CSSgetr8 r r maxr' r< r r rz rs ) tflite_inputinput_is_filepath file_handle file_datar) rG toplevel_stuffkeymappingbuffer_keys_to_displayoperator_keys_to_displayr+ re rf
tensor_mapperrg op_keys_to_displaytensor_keys_to_displays r create_htmlr sl
77>>, ' . = >> Y'<+@+@+H d # 2{ k . . 01 2
%i
0d w '
YYtL)
*d > ?? #L 1D $$,$
**$%6\ z+. +$$ Plc7 g3
8N OODP ,$ $^%5 67 -/@ A ,.> ? / 1
! MaAn -q1J/KLAnM #4 #45 $olA $$D OM &M#]3i 5O3)=9; &'7 8 %'7 8/ 79I 4 6
$| 33D ''D H+Y< M "Y
$>?,1
3 3D D a l,B CCD ~
d
+0BCCd EI
DM,= 99DHDI$N
$
DO-C
DD$
%%$
D!1 24L
MM$
$
+i2 2s I++I5c | d }| d }t | }t |d 5 }|j | d d d y # 1 sw Y y xY w# t $ r t d| d z Y y w xY w)NrU rZ wz&Usage: %s