Core ML Models

Source: https://github.com/likedan/Awesome-CoreML-Models at f65751092d0b
License: MIT
Read the English original

This edition is an AI-assisted translation.

自 iOS 11 起，Apple 发布了 Core ML 框架，以帮助开发者将机器学习模型集成到应用中。官方文档

我们整理了规模最大的 Core ML 格式机器学习模型集合，以帮助 iOS、macOS、tvOS 和 watchOS 开发者试验机器学习技术。

如果你已经转换了一个 Core ML 模型，欢迎提交拉取请求。

最近，我们加入了可视化工具。其中一个是 Netron。

模型

图像 - 元数据/文本

接收图像数据作为输入，并输出关于该图像的有用信息的模型。

• TextDetection - 使用 Vision 内置模型实时检测文本。 Download | Demo | Reference
• PhotoAssessment - 使用 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
• Car Recognition - 预测汽车的品牌与型号。 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
• Nudity - 将图像分类为 NSFW（裸体）或 SFW（非裸体） Download | Demo | Reference
• TextRecognition (ML Kit) - 使用 ML Kit 内置模型实时识别文本。 Download | Demo | Reference
• ImageSegmentation - 将相机画面或图像的像素分割为一组预定义的类别。 Download | Demo | Reference
• DepthPrediction - 从单张图像预测深度。 Download | Demo | Reference

图像 - 图像

对图像进行变换的模型。

• HED - 从彩色图像中检测嵌套边缘。 Download | Demo | Reference
• AnimeScale2x - 处理双三次插值的动漫风格艺术作品 Download | Demo | Reference

文本 - 元数据/文本

处理文本数据的模型

• Sentiment Polarity - 根据句子预测正面或负面情绪。 Download | Demo | Reference
• DocumentClassification - 将新闻文章分类为 5 个类别之一。 Download | Demo | Reference
• iMessage Spam Detection - 检测一条消息是否为垃圾消息。 Download | Demo | Reference
• NamesDT - 使用 DecisionTreeClassifier 进行性别分类 Download | Demo | Reference
• Personality Detection - 根据用户文档（句子）预测性格。 Download | Demo | Reference
• BERT for Question answering - BERT 问答的 Swift Core ML 3 实现 Download | Demo | Reference
• GPT-2 - OpenAI GPT-2 文本生成（Core ML 3） Download | Demo | Reference

杂项

• Exermote - 当 iPhone 佩戴在右大臂上时，预测正在进行的运动。 Download | Demo | Reference
• GestureAI - 根据给定位置和流派推荐艺术家。 Download | Demo | Reference
• Artists Recommendation - 根据给定位置和流派推荐艺术家。 Download | Demo | Reference
• ChordSuggester - 根据输入的和弦进行预测最有可能的下一个和弦。 Download | Demo | Reference

语音处理

• Streaming ASR – 面向 iOS 的实时流式语音识别引擎。使用 Fast Conformer + CTC，完全在设备端运行。
  Download | Demo | Reference
• Keyword Spotting (KWS) – 基于轻量级 CRNN 架构的端侧关键词唤醒引擎，针对移动设备优化。
  Download | Demo | Reference

可视化工具

帮助可视化 CoreML 模型的工具

• Netron

支持的格式

可转换为 Core ML 的模型格式列表及示例

• Caffe
• Keras
• XGBoost
• Scikit-learn
• MXNet
• LibSVM
• Torch7

黄金收藏

可转换为 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 单图像地平线估计。

贡献与许可

• 查看指南
• 基于 MIT 许可分发。更多详情请参阅 LICENSE。