freeze graph problem - Githubissues

microsoft / Recursive-Cascaded-Networks

[ICCV 2019] Recursive Cascaded Networks for Unsupervised Medical Image Registration

https://arxiv.org/abs/1907.12353

MIT License

361 stars 87 forks source link

freeze graph problem #65

Open 17dacheng opened 2 years ago

17dacheng commented 2 years ago

hi, author, when i freeze the model, what should be set as output_node_names? for example, if i use 'save/SaveV2' and generate a .pb model. when i load .pb model, it will prompt message 'voxel1' do not exist, this should be a input placeholder in the graph. but i cannot print it out when loading .pb. pls help and thanks in advance

zsyzzsoft commented 2 years ago

You can print out all the variable names via tensorflow api.