我想在 tensorflowjs 中使用 Niftynet zoo 中的模型。
当我从 Niftynet zoo 下载模型时，它们是检查点格式 (ckpt)。但是，tensorflowjs 期望模型为 pb 格式。此外，检查点格式不包含图形信息，因此我们需要使用 simple_save 或类似的 API [1] 保存模型。
所以我打算做的是：
- 从 Niftynet 动物园下载模型
- 恢复它。（这一步我需要帮助）
- 使用 simple_save api 保存完整的模型。这将产生 .pb 模型。（这一步我需要帮助）
- 使用 tensorflowjs-converter [2] 将其转换为 tensorflowjs 可以使用的 pb 格式。
那么来自 niftynet 社区的人可以帮助我完成第 2 步和第 3 步：如何使用检查点文件和配置文件加载模型并使用 simple_save api 完全保存它。