python - 你能用 Swift 直接使用 python 制作的 tensorflow 保存模型吗？

Question

我已经训练了一个 tensorflow AI 并保存了模型以使用 python 进行预测。我想将该模型移植到 SWIFT。开箱即用是否兼容？（比如导入模型保存的文件）还是我们应该重新编码训练过程？

标签： pythonswifttensorflowartificial-intelligence

Answer 1

据我了解，截至 2019 年 12 月下旬至 2020 年 1 月上旬，在 Swift for TensorFlow (S4TF) 中本地保存/加载模型的工作正在进行中。正如@dynamicwebpaige 最近在 S4TF Google Group讨论中所说，该团队正在努力实现 TensorFlow C++ API，主要专注于使用 ETA 1Q 2020 进行 SavedModel 推理。在同一讨论中，一位用户分享了 S4TF > Keras（通过 Python 互操作） > CoreML（适用于 Apple 应用程序）演示链接。此外，还有一个您可能会觉得有用的演示文稿。