培训机器学习算法的密集程度如何？

Question

我想使用iOS的新CoreML框架制作应用程序来进行图像识别。要做到这一点，我可能需要训练自己的模型，我想知道它需要多少数据和计算能力。这是我在使用谷歌图像获取源数据的双核i5 Macbook Pro上可以实现的，还是会更多参与？

Answer 1

这取决于您希望训练模型识别的图像类型。

经常做的是微调现有模型。你需要一个预训练版的Inception-v3（让我们说），然后用你自己的替换最后一层。你在自己的图像上训练最后一层。

您仍然需要相当数量的培训图片（每个类别只有100个，但更多更好），但您可以在30分钟到几个小时之间的任何时间在MacBook Pro上执行此操作。

TensorFlow带有一个script，可以很容易地做到这一点。 Keras有great blog post如何做到这一点。我使用TensorFlow脚本重新训练Inception-v3以区分我的两只猫，每只猫的50张图像。

如果您想从头开始训练，您可能希望使用AWS，Google的Cloud ML Engine或像FloydHub这样容易的事情在云端进行此操作。