继续训练CoreML模型

时间:2020-06-11 09:15:43

标签: tensorflow coreml turi-create

我试图更好地了解如何在Turi Create中创建对象检测模型(用于CoreML)。我正在尝试创建一个模型,以检测我设计和打印的自定义图像。为了避免拍摄大量照片,我认为我会使用Turi Create提供的one-shot-object-detection功能。到现在为止还挺好。我向算法提供了两个起始图像,并成功生成了合成数据集并创建了一个可靠的模型。

现在我想知道当我想添加第三类时会发生什么。我当然可以添加第三个启动程序映像,然后再次运行代码,但这感觉2/3的工作是多余的...

是否有办法继续训练以前训练过的模型,或者组合多个模型,这样我每次添加类别时都不必从头开始重新训练模型?如果不是,是否还有其他方法可以完成此操作(例如TensorFlow)?

1 个答案:

答案 0 :(得分:1)

Turi Create提供的再培训选项相当有限(基本上没有)。如果您想对过程进行更多控制,那么使用TensorFlow之类的工具是更好的选择。