训练机器学习算法的强度



我想使用iOS的新CoreML框架制作一个应用程序,该框架可以进行图像识别。为此,我可能必须训练自己的模型,我想知道它到底需要多少数据和计算能力。这是我可以使用Google Images作为源数据的双核i5 Macbook Pro上可行的完成的,还是会涉及更多?

这取决于您要训练模型识别哪种类型的图像。

通常做的是微调现有模型。你采用 Inception-v3 的预训练版本(比方说(,然后用你自己的层替换最后一层。您可以在自己的图像上训练最后一层。

您仍然需要相当数量的训练图像(每个类别有 100 个,但越多越好(,但您可以在 MacBook Pro 上完成此操作,时间在 30 分钟到几个小时之间。

TensorFlow附带了一个脚本,使做到这一点变得非常容易。Keras有一篇关于如何做到这一点的很棒的博客文章。我使用 TensorFlow 脚本重新训练 Inception-v3,从每只猫的 50 张左右的图像中分辨出我的两只猫。

如果你想从头开始训练,你可能想使用AWS,谷歌的Cloud ML引擎或像FloydHub这样简单的东西在云中做到这一点。

相关内容

  • 没有找到相关文章

最新更新