如何解决如何将 TensorFlow pb/ckpt 模型切成两部分?/ 如何去除 TensorFlow pb 模型的前端节点
我正在使用 Tensorflow 对象检测 API(tf1.15) 来训练我的人脸检测模型。 TF API 易于使用且其推理模型具有良好的结构,即
[preprocess]-[feather_extractor]-[predict]-[postprocess]
现在我需要转换 pb 模型以适合我的嵌入式系统设备。但是 pb 模型中的某些 op(TensorArrayGatherV3)
与设备不匹配。我找到了它们在 [preprocess]
和 [postprocess]
模块中的操作。所以我想删除这两个模块并使用 [feather_extractor]-[predict]
作为单个推理 pb 模型。
我使用 graph_util.convert_variables_to_constants
成功移除了后部模块 [postprocess]
。
但是,我找不到删除前端模块 [preprocess]
的方法。如何删除前端节点?我使用 tensorflow.export_model
,所以我有 ckpt/pb/saved_model
。
请帮我解决这个问题。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。