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.

Depuis iOS 11, Apple a publié le framework Core ML pour aider les développeurs à intégrer des modèles d’apprentissage automatique dans leurs applications. La documentation officielle

Nous avons rassemblé la plus grande collection de modèles d’apprentissage automatique au format Core ML, afin d’aider les développeurs iOS, macOS, tvOS et watchOS à expérimenter les techniques d’apprentissage automatique.

Si vous avez converti un modèle Core ML, n’hésitez pas à soumettre une pull request.

Récemment, nous avons ajouté des outils de visualisation. En voici un : Netron.

Modèles

Image - Métadonnées/Texte

Modèles qui prennent des données image en entrée et produisent des informations utiles à propos de l’image.

• TextDetection - Détection de texte en temps réel à l’aide du modèle intégré Vision. Download | Demo | Reference
• PhotoAssessment - Évaluation de photos à l’aide de Core ML et Metal. Download | Demo | Reference
• PoseEstimation - Estimation de la pose humaine à partir d’une image, pour mobile. Download | Demo | Reference
• MobileNet - Détecte les objets dominants présents dans une image. Download | Demo | Reference
• Places CNN - Détecte la scène d’une image parmi 205 catégories comme chambre, forêt, côte, etc. Download | Demo | Reference
• Inception v3 - Détecte les objets dominants présents dans une image. Download | Demo | Reference
• ResNet50 - Détecte les objets dominants présents dans une image. Download | Demo | Reference
• VGG16 - Détecte les objets dominants présents dans une image. Download | Demo | Reference
• Car Recognition - Prédit la marque et le modèle d’une voiture. Download | Demo | Reference
• YOLO - Reconnaît les objets présents dans une image et leur position. Download | Demo | Reference
• AgeNet - Prédit l’âge d’une personne à partir de son portrait. Download | Demo | Reference
• GenderNet - Prédit le sexe d’une personne à partir de son portrait. Download | Demo | Reference
• MNIST - Prédit les chiffres manuscrits (dessinés) à partir d’images. Download | Demo | Reference
• EmotionNet - Prédit l’émotion d’une personne à partir de son portrait. Download | Demo | Reference
• SentimentVision - Prédit un sentiment positif ou négatif à partir d’images. Download | Demo | Reference
• Food101 - Prédit le type d’aliments à partir d’images. Download | Demo | Reference
• Oxford102 - Détecte le type de fleurs à partir d’images. Download | Demo | Reference
• FlickrStyle - Détecte le style artistique des images. Download | Demo | Reference
• RN1015k500 - Prédit le lieu où une photo a été prise. Download | Demo | Reference
• Nudity - Classe une image comme NSFW (nu) ou SFW (non nu) Download | Demo | Reference
• TextRecognition (ML Kit) - Reconnaissance de texte en temps réel à l’aide du modèle intégré ML Kit. Download | Demo | Reference
• ImageSegmentation - Segmente les pixels d’une image ou d’une caméra en un ensemble prédéfini de classes. Download | Demo | Reference
• DepthPrediction - Prédit la profondeur à partir d’une seule image. Download | Demo | Reference

Image - Image

Modèles qui transforment des images.

• HED - Détecte les contours imbriqués à partir d’une image couleur. Download | Demo | Reference
• AnimeScale2x - Traite une œuvre de style anime mise à l’échelle par bicubique Download | Demo | Reference

Texte - Métadonnées/Texte

Modèles qui traitent des données textuelles

• Sentiment Polarity - Prédit un sentiment positif ou négatif à partir de phrases. Download | Demo | Reference
• DocumentClassification - Classifie les articles de presse dans l’une de 5 catégories. Download | Demo | Reference
• iMessage Spam Detection - Détecte si un message est un spam. Download | Demo | Reference
• NamesDT - Classification du sexe à l’aide de DecisionTreeClassifier Download | Demo | Reference
• Personality Detection - Prédit la personnalité à partir de documents utilisateur (phrases). Download | Demo | Reference
• BERT for Question answering - Implémentation Swift Core ML 3 de BERT pour la question-réponse Download | Demo | Reference
• GPT-2 - Génération de texte OpenAI GPT-2 (Core ML 3) Download | Demo | Reference

Divers

• Exermote - Prédit l’exercice lorsque l’iPhone est porté sur le bras droit. Download | Demo | Reference
• GestureAI - Recommande un artiste selon un lieu et un genre donnés. Download | Demo | Reference
• Artists Recommendation - Recommande un artiste selon un lieu et un genre donnés. Download | Demo | Reference
• ChordSuggester - Prédit l’accord suivant le plus probable selon la progression d’accords saisie. Download | Demo | Reference

Traitement de la parole

• Streaming ASR – Moteur de reconnaissance vocale en flux temps réel pour iOS. Utilise Fast Conformer + CTC, fonctionne entièrement sur l’appareil.
  Download | Demo | Reference
• Keyword Spotting (KWS) – Moteur de détection de mots-clés sur appareil utilisant une architecture CRNN légère, optimisé pour les mobiles.
  Download | Demo | Reference

Outils de visualisation

Outils qui aident à visualiser les modèles CoreML

• Netron

Formats pris en charge

Liste des formats de modèles convertibles en Core ML, avec exemples

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

Le Graal

Collections de modèles d’apprentissage automatique convertibles en Core ML

• Caffe Model Zoo - Grande liste de modèles au format Caffe.
• TensorFlow Models - Modèles pour TensorFlow.
• TensorFlow Slim Models - Une autre collection de modèles TensorFlow.
• MXNet Model Zoo - Collection de modèles MXNet.

Modèles d’apprentissage automatique individuels convertibles en Core ML. Nous ajusterons la liste au fur et à mesure de leur conversion.

• LaMem Note la mémorabilité des images.
• ILGnet Évaluation esthétique des images.
• Colorization Colorisation automatique à l’aide de réseaux de neurones profonds.
• Illustration2Vec Estime un ensemble de tags et extrait des vecteurs de caractéristiques sémantiques à partir d’illustrations données.
• CTPN Détection de texte dans les images naturelles.
• Image Analogy Trouve des correspondances denses sémantiquement significatives entre deux images d’entrée.
• iLID Identification automatique de la langue parlée.
• Fashion Detection Détection de vêtements dans les images.
• Saliency La prédiction des zones saillantes dans les images a traditionnellement été traitée avec des caractéristiques conçues à la main.
• Face Detection Détecte un visage dans une image.
• mtcnn Détection et alignement conjoints de visages.
• deephorizon Estimation de la ligne d’horizon à partir d’une seule image.

Contribution et licence

• Voir le guide
• Distribué sous licence MIT. Voir LICENSE pour plus d’informations.