Freeze model

[l0stpenguin]

Hi,
I am interested in the depth map estimation of this project. I tried freezing the model to a .pb file but i cannot figure out the output node which produces the depth map. Can you please tell me the output node name?

checkpoint_file = './model/deeplens/deeplens.ckpt'
frozen_model_file = './deeplens.pb'
output_node_names = '??' 
# Call freeze graph
freeze_graph(
            input_graph="",
             input_saver="",
             input_binary=False,
             input_checkpoint=checkpoint_file,
             output_node_names=output_node_names,
             restore_op_name="save/restore_all",
             filename_tensor_name="save/Const:0",
             output_graph=frozen_model_file,
             clear_devices=True,
             initializer_nodes="")

[jdluzen]

Hi @l0stpenguin, did you get anywhere with this? I've also been trying a variety of ways to freeze and/or export this graph for many days. I've gotten an output file, but I don't think it's correct because it has over 4000 nodes in it, and I get an error when trying to import it back in.