RotationDetection
RotationDetection copied to clipboard
子项目rsdet中模型转换成pb格式找不到output_node_names
使用子项目rsdet成功训练出ckpt模型之后,我需要把模型转换成pb格式,查看代码发现输出节点应该是在命名“postprocess_detctions”中,但我使用tensorboard查看计算图graph时,没有发现该输出节点,打印ckpts所有节点也没有发现该节点。请问我如何才能找到模型的输出节点名称output_node_names,并且生成pb格式的权重呢? 感谢,期待您的回复!