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。