Guest User

Untitled

a guest
Oct 17th, 2018
94
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.04 KB | None | 0 0
  1. def describe_graph(graph_def, show_nodes=False):
  2. print('Input Feature Nodes: {}'.format(
  3. [node.name for node in graph_def.node if node.op=='Placeholder']))
  4. print('')
  5. print('Unused Nodes: {}'.format(
  6. [node.name for node in graph_def.node if 'unused' in node.name]))
  7. print('')
  8. print('Output Nodes: {}'.format(
  9. [node.name for node in graph_def.node if 'predictions' in node.name]))
  10. print('')
  11. print('Quantization Nodes: {}'.format(
  12. [node.name for node in graph_def.node if 'quant' in node.name]))
  13. print('')
  14. print('Constant Count: {}'.format(
  15. len([node for node in graph_def.node if node.op=='Const'])))
  16. print('')
  17. print('Variable Count: {}'.format(
  18. len([node for node in graph_def.node if 'Variable' in node.op])))
  19. print('')
  20. print('Identity Count: {}'.format(
  21. len([node for node in graph_def.node if node.op=='Identity'])))
  22. print('', 'Total nodes: {}'.format(len(graph_def.node)), '')
  23.  
  24. if show_nodes==True:
  25. for node in graph_def.node:
  26. print('Op:{} - Name: {}'.format(node.op, node.name))
Add Comment
Please, Sign In to add comment