我已经训练了一个TensorFlow AI模型,并保存了该模型以便在Python中进行预测。我希望将该模型移植到Swift中。模型是否可以直接兼容(例如导入保存的模型文件),还是需要重新编写训练过程?
回答:
根据我的了解,截至2019年12月末至2020年1月初,Swift for TensorFlow(S4TF)中的模型保存/加载功能正在开发中。正如@*** 在最近的S4TF Google Group 讨论中所说,团队正在努力实现一个TensorFlow C++ API,主要关注SavedModel推理,预计2020年第一季度完成。在同一讨论中,一位用户分享了一个从S4TF到Keras(通过Python互操作)再到CoreML(用于Apple应用)的演示链接。此外,还有一个您可能会觉得有用的演示文稿。