tensorflow - 如何将 saved_model.pb 转换为 EvalSavedModel？

Question

我正在浏览评估 TensorFlow 模型的 tensorflow-model-analysis 文档。入门指南讨论了一个名为 EvalSavedModel 的特殊 SavedModel。

引用入门指南:

This EvalSavedModel contains additional information which allows TFMA to compute the same evaluation metrics defined in your model in a distributed manner over a large amount of data, and user-defined slices.

我的问题是如何将已经存在的 saved_model.pb 转换为 EvalSavedModel？

Answer 1

EvalSavedModel 导出为 SavedModel 消息，因此不需要进行此类转换。

EvalSavedModel 在底层使用 SavedModelBuilder。它用几个占位符填充估计器图，创建一些额外的度量集合。稍后，它执行简单的 SavedModelBuilder 过程。

来源 - https://github.com/tensorflow/model-analysis/blob/master/tensorflow_model_analysis/eval_saved_model/export.py#L228

附言我想你想在你的模型上运行 model-analysis，由 SavedModelBuilder 导出。由于 SavedModel 既没有度量节点也没有相关集合，它们是在 EvalSavedModel 中创建的，所以这样做是没有用的 - model-analysis 只是根本找不到与您的估算器相关的任何指标。