Given a tensor flow model graph, how to find the input node and output node names

by

Try this:

run python

>>> import tensorflow as tf
>>> gf = tf.GraphDef()
>>> gf.ParseFromString(open('/your/path/to/graphname.pb','rb').read())

and then

>>> [n.name + '=>' +  n.op for n in gf.node if n.op in ( 'Softmax','Placeholder')]

Then, you can get result similar to this:

['Mul=>Placeholder', 'final_result=>Softmax']

But I’m not sure it’s the problem of node names regarding the error messages.
I guess you provided wrong arguements when loading the graph file or your generated graph file is something wrong?

Check this part:

E/AndroidRuntime(16821): java.lang.IllegalArgumentException: Incompatible 
shapes: [1,224,224,3] vs. [32,1,1,2048]

UPDATE:
Sorry,
if you’re using (re)trained graph , then try this:

[n.name + '=>' +  n.op for n in gf.node if n.op in ( 'Softmax','Mul')]

It seems that (re)trained graph saves input/output op name as “Mul” and “Softmax”, while optimized and/or quantized graph saves them as “Placeholder” and “Softmax”.

BTW, using retrained graph in mobile environment is not recommended according to Peter Warden’s post: https://petewarden.com/2016/09/27/tensorflow-for-mobile-poets/ . It’s better to use quantized or memmapped graph due to performance and file size issue, I couldn’t find out how to load memmapped graph in android though…:(
(no problem loading optimized / quantized graph in android)

More Related Contents:

  1. Problems in Android repository Flutter TensorFlow-lite by Bintray 502
  2. How do I disable orientation change on Android?
  3. Load files bigger than 1M from assets folder
  4. Self-signed SSL acceptance on Android
  5. Android new Bottom Navigation bar or BottomNavigationView
  6. error opening trace file: No such file or directory (2)
  7. Read only file system on Android
  8. Android studio isn’t opening on Windows
  9. Change Background color of the action bar using AppCompat
  10. Could not find com.google.android.gms:play-services:3.1.59 3.2.25 4.0.30 4.1.32 4.2.40 4.2.42 4.3.23 4.4.52 5.0.77 5.0.89 5.2.08 6.1.11 6.1.71 6.5.87
  11. Android – custom UI with custom attributes
  12. Changing the background color of a Tab in TabLayout (Android design support library) doesn’t occupy the entire tab space
  13. Sending intent to BroadcastReceiver from adb
  14. Why “This app has been built with an incorrect configuration” error occured in some phones?
  15. Android SDK is missing, out of date, or is missing templates. Please ensure you are using SDK version 22 or later
  16. How to create a DialogFragment without title?
  17. Change fill color on vector asset in Android Studio
  18. How to pass parameters to OnClickListener?
  19. Difference between iPhone Simulator and Android Emulator
  20. RecyclerView items with big empty space after 23.2.0
  21. Android Studio Project Structure (v.s. Eclipse Project Structure)
  22. Getting Header from Response (Retrofit / OkHttp Client)
  23. How to make apk Secure. Protecting from Decompile
  24. How does doze mode affect background/foreground services, with/without partial/full wakelocks?
  25. How to access resource with dynamic name in my case?
  26. React-native google signin gives Developer Error
  27. Android: How to check if a rectangle contains touched point?
  28. onKeyDown in a service? (Global Hot Keys)
  29. When to use the attach and detach methods of FragmentTransaction
  30. Letting WebView on Android work with prefers-color-scheme: dark

Leave a Comment