o
2hfE @ s d Z ddlZddlZddlZddlZddlZeje d
ej dds-ddl m
Z nddlm
Z dZdZdd Zd
d Zd d
ZG dd dZG dd dZG dd dZdd Zdd Zd&ddZdd Zd d Z d d Z d&d!d"Z d#d$ Z e d%kre ej! dS dS )'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 C * t jj D ] \}}|| kr| S qdS )z4Converts a numerical enum to a readable tensor type.N) schema_fb
TensorType__dict__items) tensor_typenamevalue r
Z/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/lite/tools/visualize.pyTensorTypeToName
r c C r )z3Converts a builtin op code enum to a readable name.N)r BuiltinOperatorr r )coder r r
r
r BuiltinCodeToName r r c C s8 t | tr| S d}| dur| D ]
}|tt| }q|S )z;Converts a list of integers to the equivalent ASCII string. N)
isinstancestrchrint) name_listresultvalr
r
r NameListToString s
r c @ e Zd ZdZdd Zdd ZdS ) OpCodeMapperz#Maps an opcode index to an op name.c C sR i | _ t|d D ] \}}t|d | j |< | j | dkr&t|d | j |< q d S )Noperator_codes builtin_codeCUSTOM custom_code) code_to_name enumerater r )selfdataidxdr
r
r __init__ s zOpCodeMapper.__init__c C s& || j vrd}n| j | }d||f S )Nz z%s (%d))r# )r% xsr
r
r __call__ s
zOpCodeMapper.__call__N__name__
__module__ __qualname____doc__r) r, r
r
r
r r s r c @ s e Zd ZdZdd ZdS )DataSizeMapperz(For buffers, report the number of bytes.c C s |d ur
dt | S dS )Nz%d bytesz--)len)r% r* r
r
r r, s zDataSizeMapper.__call__N)r. r/ r0 r1 r, r
r
r
r r2 s r2 c @ r ) TensorMapperzGMaps a list of tensor indices to a tooltip hoverable indicator of more.c C s
|| _ d S N)r& )r%
subgraph_datar
r
r r) s
zTensorMapper.__init__c C s d}|d u r|S |d7 }|D ]C}| j d | }|t|d 7 }|t|d d 7 }|t|d d 7 }|d|v r>t|d nd7 }|d |v rLt|d ndd
7 }q|d 7 }|t|7 }|d 7 }|S ) Nr z3tensors r typeshapez[]shape_signaturez z)r& r r r repr)r% r* htmlitensorr
r
r r, s& zTensorMapper.__call__Nr- r
r
r
r r4 s r4 c C s | sdS g }| D ]C\}}t|}t|tr?t|dkr?|dd }|dd }ddtt| ddtt| d }nt|}|| d
| q
d d| d S )
zBPretty-print the quantization dictionary, truncating large arrays.r N
i[z, z, ..., ]z: {})r r< r listr3 joinmapappend)q items_strkeyr key_strheadtail value_strr
r
r QuantizationMapper) s
rQ c C s dd }dd }g }g }i }i }d} d}
t |d pg D ]{\} } | d d urQt | d D ]$\}
}||vrD| d
d | |
d |
f||< ||||| d q,| d
d urt | d
D ]$\}}||vru| d
d | |d |
f||< ||||| d q]||| || d d| | d | d q t |d D ]/\}}||v r|| n ||v r|| nd}|||dt|dg |f d |d |d d qt||d}t|| f }|S )zAProduces the HTML required to have a d3 visualization of the dag.c S d| S )Nzt%dr
r' r
r
r
TensorNameA z!GenerateGraph..TensorNamec S rR )Nzo%dr
rS r
r
r OpNameD rU z GenerateGraph..OpName operatorsinputsNg ? )sourcetargetoutputs)r] r\ opcode_index )idr groupr* yr7 )r r z%r (%d)r: r )nodesedges)r$ rI getattrjsondumps_D3_HTML_TEMPLATE) subgraph_idxg
opcode_mapperrT rV re rd firstsecond
pixel_mult
width_multop_indexoptensor_input_position tensor_indextensor_output_positionr? initial_y graph_strr= r
r
r
GenerateGraph> sd
rx Tc C s d}|d7 }|d7 }|r|d7 }|D ]
\}}|d| 7 }q|d7 }t | D ]5\}}|d7 }|r5|d| 7 }|D ] \}}||v rC|| nd}|du rK|n||}|d | 7 }q7|d7 }q%|d
7 }|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
index
z
%s
z
z
%d
Nz,
%s
)r$ ) r
keys_to_print
display_indexr= hmapperr' r? r r
r
r GenerateTableHtmlz s&
r c C s t dd| }t dd| 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_inputs1r
r
r CamelCaseToSnakeCase s r c s t | tst | tst | tr| S t| drCi }t| D ]$}| |}t|s@|d dkr@t|}|dkr7dn }t ||||< q |S t | t
j rQ rM| S | S t| dr_ fdd| D S | S ) 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__c s g | ]}t | qS r
)FlatbufferToDict).0entrypreserve_as_numpyr
r
s z$FlatbufferToDict..)
r r floatr hasattrdir__getattribute__callabler r npndarraytolist)fbr r attribute_name attribute
snake_namepreserver
r r r s"
r c C s& t j| d}t j|}t|ddS )Nr Fr )r ModelGetRootAsModelModelT InitFromObjr ) buffer_data model_objmodelr
r
r CreateDictFromFlatbuffer s r c C s |rIt j| std| | ds| dr8t| d}t| }W d n1 s.w Y t|}n| drEt
t| }ntdt| }d}|t 7 }|d 7 }|r[| nd
|d < g d }|d
7 }|D ]\}}|ssdd }|| |} |d|| f 7 }qi|d7 }dt
fg}
dtfdtfdg} |d D ]
} t| d | d | d< qt|d D ]}\}
}|d7 }t|}t|}d|fd|fd d |fg}d tfd tfd d!d"d#tfg}|d$|
7 }|d%7 }|t|d |d d&gd|fd|fgd'd(7 }|d)7 }|t|d* |7 }|d+ r|d,7 }|t|d+ |7 }|d-|
f 7 }|t|
||7 }|d.7 }q|d/7 }|t|d0 |
7 }|d17 }|t|d | 7 }|d27 }|S )3aH Returns html description with the given tflite model.
Args:
tflite_input: TFLite flatbuffer model path or model object.
input_is_filepath: Tells if tflite_input is a model path or a model object.
Returns:
Dump of the given tflite model in HTML format.
Raises:
RuntimeError: If the input is not valid.
zInvalid filename %rz.tflitez.binrbNz.jsonz#Input file was not .tflite or .jsonr z
TensorFlow Lite ModelzNull (used model object)filename))r NversionN) descriptionNry c S s | S r5 r
)r* r
r
r s z create_html..z@
%s
%s
rz r& r r" r r deprecated_builtin_code subgraphsz
rZ r^ )builtin_optionsNr_ r r9 )r: N)r; N)bufferN quantizationz
Subgraph %d
z
Inputs/Outputs
)rZ r^ F)r| z
Tensors
r7 rY z
Ops
z6
z
z
Buffers
r z
Operator Codes
z
)ospathexists RuntimeErrorendswithopen bytearrayreadr rg load_CSSgetr2 r r maxr$ r4 r r rQ r rx ) tflite_inputinput_is_filepath file_handle file_datar& r= toplevel_stuffrL mappingr buffer_keys_to_displayoperator_keys_to_displayr( rj rk
tensor_mapperrl op_keys_to_displaytensor_keys_to_displayr
r
r create_html s
r c C s| z
| d }| d }W n t y td| d Y d S w t|}t|d}|| W d d S 1 s7w Y d S )Nr[ r` z&Usage: %s