SCNN输入输出节点的问题

cardwing / Codes-for-Lane-Detection

Learning Lightweight Lane Detection CNNs by Self Attention Distillation (ICCV 2019)

MIT License

1.04k stars 333 forks source link

SCNN输入输出节点的问题 #117

Closed ryl1122 closed 5 years ago

ryl1122 commented 5 years ago

我在从ckpt转pb的时候，以及pb转C代码时，转换的脚本需要明确输入输出的节点。我尝试了很多节点，但是都不对。请问下这个输入输出的节点如何查看？

ycdhqzhiai commented 5 years ago

你好，你输入输出节点选取的是什么啊，我输出节点选择的是lanenet_loss/inference/encode/relu6，还没有测试，不知道你试过没有

cardwing commented 5 years ago

Have a try of the following codes: checkpoint_dir = 'path/to/culane_lanenet_vgg_2018-10-24-22-40-59.ckpt-17000' for var_name, _ in tf.contrib.framework.list_variables(checkpoint_dir): # Load the variable var = tf.contrib.framework.load_variable(checkpoint_dir, var_name) print(var_name)

ycdhqzhiai commented 5 years ago

@ryl1122 可以参考下https://github.com/ycdhqzhiai/SCNN_PB