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, смело отправляйте pull request.

Недавно мы добавили инструменты визуализации. Вот один из них — 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.