Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix #447: Changed format for exporting models to MetaGraphDef for Tensorflow #455

Open
wants to merge 9 commits into
base: master
Choose a base branch
from
+14,562 −27
Open

Fix #447: Changed format for exporting models to MetaGraphDef for Tensorflow #455

Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
2fe5a5d
Make celery export models as metagraphdef text file
abhigyan7 Nov 15, 2018
a1e0eb6
Updated documentation to use .meta files
abhigyan7 Nov 15, 2018
b2d292d
Add support for importing metagraphdefs and binary graphdefs
abhigyan7 Nov 18, 2018
ec163b2
Merge branch 'master' into tf_metagraphdef
abhigyan7 Nov 18, 2018
ad21050
Export support for json2pbtxt.py
abhigyan7 Nov 19, 2018
02df885
Renamed tensorflow_app/views/json2pbtxt.py -> json2meta.py
abhigyan7 Nov 19, 2018
4844984
Merge branch 'tf_metagraphdef' of https://github.com/haabhi/Fabrik in…
abhigyan7 Nov 19, 2018
179d737
Change .pbtxt to .meta in tensorflow_app export modules
abhigyan7 Nov 21, 2018
12bd6c9
Added unittest cases for importing and exporting metagraphdefs
abhigyan7 Nov 21, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Export support for json2pbtxt.py
@abhigyan7
abhigyan7 committed Nov 19, 2018
commit ad2105097ad9d26416c85d6f4be680086a62a2d7
6 changes: 3 additions & 3 deletions tensorflow_app/views/json2pbtxt.py
View file
Original file line number Diff line number Diff line change
@@ -30,6 +30,6 @@
lrn = imp.load_source('LRN', BASE_DIR + '/keras_app/custom_layers/lrn.py')
model = model_from_json(json_str, {'LRN': lrn.LRN})

sess = K.get_session()
tf.train.write_graph(sess.graph.as_graph_def(add_shapes=True), output_fld,
output_file + '.pbtxt', as_text=True)
tf.train.export_meta_graph(
os.path.join(output_fld, output_file + '.meta'),
as_text=True)