tensorflow - 区分磁盘模型的类型

Question

Tensorflow 有几种模型格式：

1. TensorFlow SavedModel 2. Frozen Model 3. Session Bundle 4. Tensorflow Hub 模块

你如何区分它们在磁盘上？（稍后与 tensorflowjs-converter 一起使用）

每个模型是如何创建的？

Answer 1

是的，有很多不同的模型类型，它们都有很好的理由。我不会声称我对每一个都非常清楚，但这就是我所知道的（我想我知道）。

• 文件：.pbPB 代表 protobuff 或 Protocol Buffer。这是模型结构，通常没有经过训练的权重，并以二进制格式存储。
• .pbtxt文件：pb供人类阅读的文件的非二进制文件。
• 未冻结的 Protobuff 文件也需要一个检查点.ckpt文件。检查点文件是所需的缺失权重集pb。
• .h5文件：模型 + Keras 保存的权重
• 该.tflite文件将是一个 TensorflowLite 模型
• 冻结模型：冻结模型结合了pb权重文件，因此您不必管理其中两个。通常，这意味着将单词添加frozen到文件名中。我确信在加载文件时可以推断出这一点，但在磁盘上它们在荣誉系统上更多一些并且没有ckpt文件。这去除了无关的图形信息；它基本上就像模型的“生产就绪”版本。
• Session Bundle：是一个目录。它们不再使用，而且很少见。
• Tensorflow Hub 模块：这些是预先存在的流行模型，很可能已经导出到 TFJS，并且不需要您手动转换它们。我认为支持它们是为了 Google 的利益，而不是我们的利益。但是很高兴知道您是否使用集线器，您可以随时转换它。

一个多导出的文件分组看起来像这个图像。从这里，你可以看到很多你可以变成 TFJS 的东西。