Awesome Models
Modelsを扱う資料や関連プロジェクトをまとめたAwesomeリストです。
画像 - メタデータ/テキスト
画像データを入力として受け取り、画像に関する有用な情報を出力するモデルです。
- TextDetection - Vision内蔵モデルを使ってリアルタイムにテキストを検出します。 ダウンロード | デモ | 参考資料
- PhotoAssessment - Core MLとMetalを使って写真を評価します。 ダウンロード | デモ | 参考資料
- PoseEstimation - モバイル環境で画像から人間の姿勢を推定します。 ダウンロード | デモ | 参考資料
- MobileNet - 画像内の主要な物体を検出します。 ダウンロード | デモ | 参考資料
- Places CNN - 寝室、森林、海岸など205カテゴリから画像の場面を検出します。 ダウンロード | デモ | 参考資料
- Inception v3 - 画像内の主要な物体を検出します。 ダウンロード | デモ | 参考資料
- ResNet50 - 画像内の主要な物体を検出します。 ダウンロード | デモ | 参考資料
- VGG16 - 画像内の主要な物体を検出します。 ダウンロード | デモ | 参考資料
- Car Recognition - 車のブランドと車種を予測します。 ダウンロード | デモ | 参考資料
- YOLO - 画像内にある物体の種類と位置を認識します。 ダウンロード | デモ | 参考資料
- AgeNet - 人物の顔写真から年齢を予測します。 ダウンロード | デモ | 参考資料
- GenderNet - 人物の顔写真から性別を予測します。 ダウンロード | デモ | 参考資料
- MNIST - 画像から手書き数字を予測します。 ダウンロード | デモ | 参考資料
- EmotionNet - 人物の顔写真から感情を予測します。 ダウンロード | デモ | 参考資料
- SentimentVision - 画像から肯定的または否定的な感情を予測します。 ダウンロード | デモ | 参考資料
- Food101 - 画像から食べ物の種類を予測します。 ダウンロード | デモ | 参考資料
- Oxford102 - 画像から花の種類を検出します。 ダウンロード | デモ | 参考資料
- FlickrStyle - 画像の芸術的スタイルを検出します。 ダウンロード | デモ | 参考資料
- RN1015k500 - 写真が撮影された場所を予測します。 ダウンロード | デモ | 参考資料
- Nudity - 画像をNSFW(ヌード)またはSFW(ヌードではない)に分類します。 ダウンロード | デモ | 参考資料
- TextRecognition (ML Kit) - ML Kit内蔵モデルを使ってリアルタイムにテキストを認識します。 ダウンロード | デモ | 参考資料
- ImageSegmentation - カメラフレームまたは画像の各ピクセルを、事前定義されたクラスへ分割します。 ダウンロード | デモ | 参考資料
- DepthPrediction - 一枚の画像から奥行きを予測します。 ダウンロード | デモ | 参考資料
画像 - 画像
画像を変換するモデルです。
- HED - カラー画像から階層的なエッジを検出します。 ダウンロード | デモ | 参考資料
- AnimeScale2x - バイキュービック拡大されたアニメ調の画像を処理します。 ダウンロード | デモ | 参考資料
テキスト - メタデータ/テキスト
テキストデータを処理するモデルです。
- Sentiment Polarity - 文から肯定的または否定的な感情を予測します。 ダウンロード | デモ | 参考資料
- DocumentClassification - ニュース記事を5カテゴリのいずれかに分類します。 ダウンロード | デモ | 参考資料
- iMessage Spam Detection - メッセージがスパムかどうかを検出します。 ダウンロード | デモ | 参考資料
- NamesDT - DecisionTreeClassifierを使って性別を分類します。 ダウンロード | デモ | 参考資料
- Personality Detection - ユーザーの文書(文章)に基づいて性格を予測します。 ダウンロード | デモ | 参考資料
- BERT for Question answering - 質問応答向けBERTのSwift Core ML 3実装です。 ダウンロード | デモ | 参考資料
- GPT-2 - OpenAI GPT-2によるテキスト生成(Core ML 3)です。 ダウンロード | デモ | 参考資料
その他
- Exermote - iPhoneを右上腕に装着したときの運動を予測します。 ダウンロード | デモ | 参考資料
- GestureAI - 指定した場所とジャンルに基づいてアーティストを推薦します。 ダウンロード | デモ | 参考資料
- Artists Recommendation - 指定した場所とジャンルに基づいてアーティストを推薦します。 ダウンロード | デモ | 参考資料
- ChordSuggester - 入力されたコード進行に基づいて、次に来る可能性が最も高いコードを予測します。 ダウンロード | デモ | 参考資料
音声処理
- Streaming ASR – iOS向けリアルタイムストリーミング音声認識エンジンです。Fast ConformerとCTCを使い、完全にデバイス上で動作します。 ダウンロード | デモ | 参考資料
- Keyword Spotting (KWS) – 軽量なCRNNアーキテクチャを使い、モバイル端末向けに最適化されたオンデバイスのキーワード検出エンジンです。 ダウンロード | デモ | 参考資料
可視化ツール
Core MLモデルの可視化を支援するツールです。
対応形式
Core MLへ変換できるモデル形式と例の一覧です。
注目コレクション
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 2枚の入力画像間で意味のある密な対応関係を見つけます。
- iLID 発話言語を自動識別します。
- Fashion Detection 画像から衣服を検出します。
- Saliency 画像内の顕著領域を予測する従来手法では、手設計の特徴量が使われてきました。
- Face Detection 画像から顔を検出します。
- mtcnn 顔検出と位置合わせを同時に行います。
- deephorizon 一枚の画像から地平線を推定します。
コントリビューションとライセンス
- ガイドを参照
- MITライセンスで配布されています。詳細はLICENSEを参照してください。