Minor fix for freeze_graph

yan99033 / monodepth-cpp

Tensorflow C++ implementation for single-image depth estimation (Monodepth)

MIT License

105 stars 33 forks source link

Minor fix for freeze_graph #3

Closed muskie82 closed 5 years ago

muskie82 commented 5 years ago

To make freeze_graph.py work, the method name has to be changed ("output" instead of "out").

yan99033 commented 5 years ago

Hey, thanks for pointing out the typo. I was supposed to use the getter method (model.get_output()) to obtain the output, which was defined in the monodepth.py.

It is now fixed and I have also updated the README.