Core ML Models

从 iOS 11 开始，苹果发布了 Core ML 框架，帮助开发者将机器学习模型集成到应用程序中. The official documentation

我们以 Core ML 格式提供了最大的机器学习模型集合，以帮助 iOS、macOS、tvOS 和 watchOS 开发人员尝试机器学习技术.

如果您已经转换了 Core ML 模型，请随时提交 pull request.

最近，我们添加了可视化工具. 这是一个 Netron.

Models

Image - Metadata/Text

以图像数据作为输入并输出有关图像的有用信息的模型. * TextDetection - 使用 Vision 内置模型实时检测文本. 下载 | Demo | Reference * 照片评估 - 使用 Core ML 和 Metal 进行照片评估. Download | Demo | Reference * PoseEstimation - 根据手机图片估计人体姿势. Download | Demo | Reference * MobileNet - 检测图像中存在的主要对象. Download | Demo | Reference * Places CNN - 从卧室、森林、海岸等 205 个类别中检测图像场景. Download | Demo | Reference * Inception v3 - 检测图像中存在的主要对象. Download | Demo | Reference * ResNet50 - 检测图像中存在的主要对象. Download | Demo | Reference * VGG16 - 检测图像中存在的主要对象. Download | Demo | Reference * 汽车识别 - 预测汽车的品牌和型号. Download | Demo | Reference * YOLO - 识别给定图像中的对象是什么以及它们在图像中的位置. Download | Demo | Reference * AgeNet - 从一个人的肖像预测一个人的年龄. Download | Demo | Reference * GenderNet - 从一个人的肖像预测一个人的性别. Download | Demo | Reference * MNIST - 从图像中预测手写（绘制）数字. Download | Demo | Reference * EmotionNet - 从一个人的肖像预测一个人的情绪. Download | Demo | Reference * SentimentVision - 从图像中预测积极或消极的情绪. Download | Demo | Reference * Food101 - 从图像中预测食物的类型. Download | Demo | Reference * Oxford102 - 从图像中检测花朵的类型. Download | Demo | Reference * FlickrStyle - 检测图像的艺术风格. Download | Demo | Reference * RN1015k500 - 预测拍摄照片的位置. Download | Demo | Reference * 裸体 - 将图像分类为 NSFW（裸体）或 SFW（非裸体） Download | Demo | Reference * TextRecognition (ML Kit) - 使用 ML Kit 内置模型实时识别文本. 下载 | Demo | Reference * 图像分割 - 将相机帧或图像的像素分割为一组预定义的类. Download | Demo | Reference * DepthPrediction - 预测单个图像的深度. Download | Demo | Reference

Image - Image

转换图像的模型. * HED - 从彩色图像中检测嵌套边缘. Download | Demo | Reference * AnimeScale2x - 处理双三次比例的动漫风格艺术品 Download | Demo | Reference

Text - Metadata/Text

处理文本数据的模型 * 情绪极性 - 从句子中预测积极或消极情绪. Download | Demo | Reference * 文档分类 - 将新闻文章分为 5 个类别之一. Download | Demo | Reference * iMessage 垃圾邮件检测 - 检测邮件是否为垃圾邮件. Download | Demo | Reference * NamesDT - 使用 DecisionTreeClassifier 进行性别分类 Download | Demo | Reference * 性格检测 - 根据用户文档（句子）预测性格. Download | Demo | Reference * 用于问答的 BERT - 用于问答的 BERT 的 Swift Core ML 3 实现 Download | Demo | Reference * GPT-2 - OpenAI GPT-2 文本生成（Core ML 3） Download | Demo | Reference

Miscellaneous

• Exermote - 当 iPhone 佩戴在右上臂时预测锻炼情况. Download | Demo | Reference
• GestureAI - 根据给定位置和流派推荐艺术家. Download | Demo | Reference
• 艺术家推荐 - 根据给定地点和流派推荐艺术家. Download | 演示 | Reference
• ChordSuggester - 根据输入的和弦进行预测最可能的下一个和弦. Download | Demo | Reference

Visualization Tools

帮助可视化 CoreML 模型的工具 * Netron

Supported formats

可转换为 Core ML 的模型格式列表及示例 * Caffe * Keras * XGBoost * Scikit-learn * MXNet * LibSVM * Torch7

The Gold

可转换为 Core ML 的机器学习模型集合

• Caffe Model Zoo - Caffe 格式的大模型列表.
• TensorFlow Models - TensorFlow 模型.
• TensorFlow Slim Models - TensorFlow 模型的另一个集合.
• MXNet Model Zoo - MXNet 模型的集合.

可以转换为 Core ML 的个人机器学习模型. 我们将在转换后不断调整列表. * LaMem 对图片的记忆力进行评分. * ILGnet 图像的审美评价. * Colorization 使用深度神经网络自动着色. * Illustration2Vec 估计一组标签并从给定插图中提取语义特征向量. * CTPN 检测自然图像中的文本. * Image Analogy 查找两个输入图像之间具有语义意义的密集对应关系. * iLID 自动口语识别. * Fashion Detection 从图像中检测布料. * Saliency 传统上，图像中显着区域的预测是通过手工制作的特征来解决的. * Face Detection 从图像中检测人脸. * mtcnn 联合面部检测和对齐. * deephorizon 单图像地平线估计.

Contributing and License

• See the guide
• 根据 MIT 许可分发. 请参阅许可证了解更多信息.