继续训练 CoreML 模型

Posted 2023-03-27

【中文标题】继续训练 CoreML 模型

【英文标题】：Continue training CoreML Model

【发布时间】：2020-09-30 21:30:21

【问题描述】：

我试图更好地了解如何在 Turi Create 中创建对象检测模型（用于 CoreML）。我正在尝试创建一个模型来检测我自己设计和打印的自定义图像。为了避免拍摄大量照片，我想我会使用 Turi Create 提供的one-shot-object-detection 功能。到目前为止，一切都很好。我为算法提供了两张起始图像，它成功地生成了合成数据集并创建了一个可靠的模型。

现在我想知道当我想添加第三个类别时会发生什么。我当然可以添加第三个起始图像并再次运行代码，但这感觉就像 2/3 的工作是多余的......

有没有办法继续训练之前训练过的模型，或者组合多个模型，这样我就不必在每次添加类别时都从头开始重新训练模型？如果没有，还有其他方法可以完成这项工作（例如 TensorFlow）吗？

【参考方案1】：

Turi Create 提供的再培训选项相当有限（基本上没有）。如果您想更好地控制流程，使用 TensorFlow 等工具是更好的选择。