动机
我在这里提供了一个简单的指南,该指南解释了从训练简单的PyTorch图像分类器到将训练后的神经网络转换为可投入生产的CoreML模型所需的步骤。 我花了几天时间浏览Internet博客,论坛和官方文档,以收集这些页面中提供的少量知识。 此仓库的真正动机是防止我忘记有关该特定主题的所有知识。 并且,如果本指南可以帮助其他人继续进行她/他的研究,那将是一个加分。 请阅读免责声明。
导游
我面临的问题非常简单。 我想知道如何在PyTorch中训练人工神经网络,以及如何将该网络转换为可在iOS应用程序中使用的CoreML模型。 简单吧?
最初,此页面中提供的指南是针对coremltools 3设计的。Apple最近发布了coremltools 4,它改变了游戏规则。 现在可以在不使用ONNX的情况下完成转换。 我本可以简单地更新coremltools 4的指南。但是,由于coremltoo
2022-05-14 19:07:56
2.2MB
Python
1