区分磁盘模型的类型

Question

Tensorflow有几个types of model formats：

  

  
1. TensorFlow SavedModel 2.冻结的模型3. Session Bundle 4. Tensorflow Hub模块
  

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

每个模型如何创建？

Answer 1

是的，有很多不同的模型类型，它们都有充分的理由。我并不是要宣称我对每个对象都非常清楚，但这是我所知道的（我想我知道）。

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

文件的多导出分组如下图所示。从这里，您可以看到很多可以变成TFJS的内容。