训练机器学习算法的强度如何？

Posted 2023-03-27

【中文标题】训练机器学习算法的强度如何？

【英文标题】：How intensive is training a machine learning algorithm?

【发布时间】：2017-11-12 12:21:45

【问题描述】：

我想使用 ios 的新 CoreML 框架制作一个应用程序来进行图像识别。为此，我可能必须训练自己的模型，并且我想知道它究竟需要多少数据和计算能力。我是否可以在双核 i5 Macbook Pro 上使用 Google 图片作为源数据来实现这一目标，还是会涉及更多？

【问题讨论】：

这取决于您使用的模型有多复杂。您将不得不尝试并找出答案。

【参考方案1】：

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

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

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

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

如果您想从头开始训练，您可能希望使用 AWS、Google 的 Cloud ML Engine 或 FloydHub 之类的简单工具在云中执行此操作。