How to convert .pb to TFLite format?

by

I am making a wild guess here, maybe you entered input_arrays=input.
Which may not be true. Use this script to find the name of the input and output arrays of the frozen inference graph

import tensorflow as tf
gf = tf.GraphDef()   
m_file = open('frozen_inference_graph.pb','rb')
gf.ParseFromString(m_file.read())

with open('somefile.txt', 'a') as the_file:
    for n in gf.node:
        the_file.write(n.name+'\n')

file = open('somefile.txt','r')
data = file.readlines()
print "output name = "
print data[len(data)-1]

print "Input name = "
file.seek ( 0 )
print file.readline()

In my case they are:

output name: SemanticPredictions
input name: ImageTensor

More Related Contents:

  1. TensorFlow and person recognition in video stream
  2. How to Properly Combine TensorFlow’s Dataset API and Keras?
  3. How Can I Define Only the Gradient for a Tensorflow Subgraph?
  4. Get the value of some weights in a model trained by TensorFlow
  5. How do I disable TensorFlow’s eager execution?
  6. Confused by the behavior of `tf.cond`
  7. TensorFlow: slow performance when getting gradients at inputs
  8. TensorFlow, why there are 3 files after saving the model?
  9. Tensorflow doesn’t seem to see my gpu
  10. tensorflow on GPU: no known devices, despite cuda’s deviceQuery returning a “PASS” result
  11. How can I run Tensorflow on one single core?
  12. Initializing tensorflow Variable with an array larger than 2GB
  13. Holding variables constant during optimizer
  14. TensorFlow: Dst tensor is not initialized
  15. Using Gensim Fasttext model with LSTM nn in keras
  16. What is the proper way to install TensorFlow on Apple M1 in 2022
  17. How do I convert a directory of jpeg images to TFRecords file in tensorflow?
  18. Tensorflow get all variables in scope
  19. Could not install packages due to an EnvironmentError: [WinError 5] Access is denied:
  20. Keras split train test set when using ImageDataGenerator
  21. CUDA_ERROR_OUT_OF_MEMORY in tensorflow
  22. Implementing contrastive loss and triplet loss in Tensorflow
  23. Confusion about keras Model: __call__ vs. call vs. predict methods
  24. How to properly use tf.metrics.accuracy?
  25. ImportError: libcudnn when running a TensorFlow program
  26. Google Storage (gs) wrapper file input/out for Cloud ML?
  27. What does batch, repeat, and shuffle do with TensorFlow Dataset?
  28. How to replace (or insert) intermediate layer in Keras model?
  29. Choosing number of Steps per Epoch
  30. Parallelism isn’t reducing the time in dataset map

Leave a Comment