如何解决如何从Tensorflow Objectdetection 2中的检查点导出Frozen_inference_graph.pb
如何将经过训练的模型导出到Frozen_inference_graph.pb而不是saved_model.pb,因为当我使用Tensorflow对象检测v2随附的exporter_main_v2.py时,它会给我一个文件夹
├─ exported-models/
└─ my_model/
├─ checkpoint/
├─ saved_model/
└─ assets/
├─ variables/
└─ saved_model.pb
└─ pipeline.config
在save_model内,我有save_model.pb,但问题是我不能单独使用它进行推理,但是我需要使用它随附的变量文件夹。这就是为什么我要问theire是否是一种将训练后的模型导出到Frozen_inference_graph.pb的方法,以便将其用于推理,而无需像TF1这样的变量文件夹。
解决方法
不幸的是,TF2 不支持export_inference_graph.py,它可以完成您所要求的确切操作——但仅适用于TF1。 exporter_main_v2.py 不会导出任何冻结图。
你可以做的是使用
冻结你的图表from tensorflow.python.tools import freeze_graph
freeze_graph.freeze_graph(...)
你也可以在你电脑的tensorflow\python\tools中找到freeze_graph.py这个文件,里面有很好的使用说明。 另外,你可以看看这个如何冻结它:freeze_graph_test 但是,TF2 不再打算生成冻结图!
编辑:可以找到很好的解决方法 here
,我们在TensorFlow模型git中具有引用以导出模型以尝试使用它进行推理-https://github.com/tensorflow/models/blob/master/research/object_detection/export_inference_graph.py
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。