Largest list of models for Core ML (for iOS 11+)

Related tags

Machine Learning ios machine-learning apple awesome caffe download model models ml awesome-list curated-list caffemodel tensorflow-models ios11 coreml core-ml coreml-framework coremltools coreml-models coreml-model
Overview

Since iOS 11, Apple released Core ML framework to help developers integrate machine learning models into applications. The official documentation

We've put up the largest collection of machine learning models in Core ML format, to help iOS, macOS, tvOS, and watchOS developers experiment with machine learning techniques.

If you've converted a Core ML model, feel free to submit a pull request.

Recently, we've included visualization tools. And here's one Netron.

Awesome PRs Welcome

Models

Image - Metadata/Text

Models that take image data as input and output useful information about the image.

Image - Image

Models that transform images.

Text - Metadata/Text

Models that process text data

Miscellaneous

Visualization Tools

Tools that help visualize CoreML Models

Supported formats

List of model formats that could be converted to Core ML with examples

The Gold

Collections of machine learning models that could be converted to Core ML

Individual machine learning models that could be converted to Core ML. We'll keep adjusting the list as they become converted.

  • LaMem Score the memorability of pictures.
  • ILGnet The aesthetic evaluation of images.
  • Colorization Automatic colorization using deep neural networks.
  • Illustration2Vec Estimating a set of tags and extracting semantic feature vectors from given illustrations.
  • CTPN Detecting text in natural image.
  • Image Analogy Find semantically-meaningful dense correspondences between two input images.
  • iLID Automatic spoken language identification.
  • Fashion Detection Cloth detection from images.
  • Saliency The prediction of salient areas in images has been traditionally addressed with hand-crafted features.
  • Face Detection Detect face from image.
  • mtcnn Joint Face Detection and Alignment.
  • deephorizon Single image horizon line estimation.

Contributing and License

  • See the guide
  • Distributed under the MIT license. See LICENSE for more information.
Comments
  • Traffic Sign Detection Benchmark

    Traffic Sign Detection Benchmark

    This model should detect and recognize live video traffic signs. If there is a speed limit fo example the app should detect and recognize this shield.

    There is a lot of sources...

    Model URL

    http://btsd.ethz.ch/shareddata/ http://benchmark.ini.rub.de/?section=gtsdb&subsection=news

    Demo URL

    http://www.vision.ee.ethz.ch/~timofter/

    opened by ihyildiz 8
  • Contributing with LocationNet model

    Contributing with LocationNet model

    Model URL

    https://s3.amazonaws.com/aws-bigdata-blog/artifacts/RN1015k500/RN1015k500.mlmodel

    Demo URL

    https://github.com/awslabs/MXNet2CoreML_iOS_sample_app

    Descriptions

    Predict the location where a picture was taken.

    Checklist

    • [x] Only one item is in this pull request
    • [x] The model info contains all the required fields
    • [x] The demo project is compilable
    • [x] Has proper reference
    opened by menant 7
  • New Model: Fast-Style-Transfer

    New Model: Fast-Style-Transfer

    License

    My Code is MIT, though the model requires an agreement for commercial use

    Summary

    I implemented Fast Style Transfer in CoreML. This is similar to Fast Neural Style, but is TensorFlow. My article explains the process.

    Model URL

    https://github.com/lengstrom/fast-style-transfer

    Demo URL

    https://medium.com/@rambossa/diy-prisma-fast-style-transfer-app-with-coreml-and-tensorflow-817c3b90dacd https://github.com/mdramos/fast-style-transfer-coreml

    Samples

    Here are some CoreML models I created in the process: https://drive.google.com/drive/folders/1CBSanBHbXC5-bJNTTk3-r1WSq56z0eKG?usp=sharing -- just include these in the ios app before running

    Checklist

    • [x] Only one item is in this issue
    • [x] The model info contains all the required fields
    • [x] The demo project is compilable (AFTER INCLUDE THE MLMODELS)
    • [?] Has proper reference
    • [x] If this model takes image as input, an image type is selected instead of multiarray (See https://developer.apple.com/wwdc17/710 if you don't know how to do this)
    opened by backnotprop 5
  • Food101 model

    Food101 model

    Sorry that I don't conform to your issue guide but my issue is just to improve an already existing model in your collection.

    I saw that you added images with the input types and the output types of each model to your README and I therefore wanted to tell you that I finally converted the Food101 CoreML model in a matter where it now accepts CVPixelBuffer instead of MLMultiArray as an input.

    Just a little update :)

    opened by ph1ps 5
  • New Model: DocumentClassification

    New Model: DocumentClassification

    This model is also included in an open-source framework that can be used for document classification.

    License

    MIT

    Summary

    • Classifies documents into one of five categories (Business, Entertainment, Politics, Sports, Technology)
    • Trained with 1,500 articles from the BBC. See "References" in README

    Model URL

    Model Link

    Demo URL

    Demo app is the NewsClassifier iOS app here

    Samples

    Screenshot from the demo app here

    Checklist

    • [x] Only one item is in this issue
    • [x] The model info contains all the required fields
    • [x] The demo project is compilable
    • [x] Has proper reference
    • [x] If this model takes image as input, an image type is selected instead of multiarray (See https://developer.apple.com/wwdc17/710 if you don't know how to do this)
    opened by toddkramer 4
  • New Model: PoseEstimation

    New Model: PoseEstimation

    License

    Apache License 2.0

    Summary

    edvardHua implements PoseEstimationForMobile estimating human pose from a picture for mobile. And I make demo for that on iOS.

    • Estimate human pose
    • Train AI Challenger dataset that is single person image dataset contain about 20,000

    Model URL

    Model URL

    Demo URL

    Demo URL

    Samples

    Input

    Image[URL]

    Output

    Heatmap[Array<Array<Array>>]

    model |cpm|hourglass -----|-----|----- output | [96, 96, 14] | [48, 48, 14]

    Checklist

    • [x] Only one item is in this issue
    • [x] The model info contains all the required fields
    • [x] The demo project is compilable
    • [x] Has proper reference
    • [x] If this model takes image as input, an image type is selected instead of multiarray (See https://developer.apple.com/wwdc17/710 if you don't know how to do this)
    opened by tucan9389 3
  • Nudity model

    Nudity model

    License

    BSD-2

    Summary

    This model detects nudity and it's score from an image. The original dataset is Yahoo's OpenNSFW.

    Model URL

    Since this is too big for Github I have a GoogleDrive link: https://drive.google.com/open?id=0B5TjkH3njRqncDJpdDB1Tkl2S2s

    Demo URL

    https://github.com/ph1ps/Nudity-CoreML

    Samples

    Input: Output: SFW (100%), NSFW (0%) Input: Image, Output: NSFW (80%), SFW(20%)

    This gives you back both NSFW and SFW score in order to make people able to decide what their personal threshold is. Let's say there is a picture with NSFW - 70%, some people might consider this as safe but others not. Therefore they can say everything from 70% and down is SFW and everything from 71% to 100% is NSFW.

    Checklist

    • [x] Only one item is in this issue
    • [x] The model info contains all the required fields
    • [x] The demo project is compilable
    • [x] Has proper reference
    • [x] If this model takes image as input, an image type is selected instead of multiarray (See https://developer.apple.com/wwdc17/710 if you don't know how to do this)
    opened by ph1ps 3
  • Adding in new CoreML Model for Sound Classification

    Adding in new CoreML Model for Sound Classification

    Adding in ESC-10 CoreML Model; trained on the ESC-10 dataset, it performs simple audio event recognition. Example project + blog post are included.

    Model URL

    Demo URL

    Descriptions

    Checklist

    • [x] Only one item is in this pull request
    • [x] The model info contains all the required fields
    • [x] The demo project is compilable
    • [x] Has proper reference
    opened by narner 2
  • NamesDT

    NamesDT

    (not the author, just stumbled across this and seems it would fit in Text Analysis here)

    License

    MIT

    Summary

    A Demo application using CoreML framework for predicting gender from first names. See Is it a boy or a girl? An introduction to Machine Learning

    Model URL

    CoreML model was converted from Scikit-learn Pipeline using coremltools python package.

    Demo URL

    NamesCoreMLDemo

    Samples

    In demo README

    Checklist

    • [x] Only one item is in this issue
    • [x] The model info contains all the required fields
    • [x] The demo project is compilable
    • [x] Has proper reference
    opened by alexcurylo 2
  • Adding image depth prediction model repo

    Adding image depth prediction model repo

    Model URL

    https://github.com/iro-cp/FCRN-DepthPrediction

    Demo URL

    https://github.com/tucan9389/DepthPrediction-CoreML

    Descriptions

    Predict the depth from a single image

    Checklist

    • [x] Only one item is in this pull request
    • [x] The model info contains all the required fields
    • [x] The demo project is compilable
    • [x] Has proper reference
    opened by tucan9389 1
  • Adding image segmentation and depth prediction model repo

    Adding image segmentation and depth prediction model repo

    Model URL

    https://developer.apple.com/machine-learning/models/

    Demo URL

    https://github.com/tucan9389/ImageSegmentation-CoreML

    | screenshot 1 | screenshot 2 | screenshot 3 | | ---- | ---- | ---- | | | | |

    Descriptions

    Segment the pixels of a camera frame or image into a predefined set of classes.

    Checklist

    • [x] Only one item is in this pull request
    • [x] The model info contains all the required fields
    • [x] The demo project is compilable
    • [x] Has proper reference
    opened by tucan9389 1
Owner
Kedan Li
Founder/CEO @ Revery.AI. Virtual Try-on Solution for Fashion E-Commerce, powered by state-of-the-art AI technology
Kedan Li
GitHub
A Swift library for creating and exporting CoreML Models in Swift

SwiftCoreMLTools A Swift Library for creating CoreML models in Swift. Work in progress This library expose a (function builder based) DSL as well as a

Jacopo Mangiavacchi 140 Dec 5, 2022
Models and examples built with TensorFlow

Welcome to the Model Garden for TensorFlow The TensorFlow Model Garden is a repository with a number of different implementations of state-of-the-art

null 74.9k Dec 29, 2022
Sample code for Core ML using ResNet50 provided by Apple and a custom model generated by coremltools.

CoreML-samples This is the sample code for Core ML using ResNet50 provided by Apple. ResNet50 can categorize the input image to 1000 pre-trained categ

Yuta Akizuki 39 Nov 11, 2022
Track List Detail App With Swift

TrackListDetailApp TrackListDetailApp is master-detail application that containsa a list of items obtained from a iTunes Search API and show a detaile

Karlo Manguiat 1 Dec 12, 2021
Mobile-ios-ml - SBB Mobile Machine Learning for iOS devices

ESTA library: Machine Learning for iOS This framework simplifies the integration

Swiss Federal Railways (SBB) 9 Jul 16, 2022
BetterMood is an iOS app that uses Tensorflow to recognize user’s emotions

BetterMood is an iOS app that uses Tensorflow to recognize user’s emotions, convert it into categories then send via our api along with the user’s date of birth and name, to end up with a emotion analyse and horoscope prediction.

Yosri 2 Sep 30, 2021
Artificial intelligence/machine learning data structures and Swift algorithms for future iOS development. bayes theorem, neural networks, and more AI.

Swift Brain The first neural network / machine learning library written in Swift. This is a project for AI algorithms in Swift for iOS and OS X develo

Vishal 331 Oct 14, 2022
Model stock prediction for iOS

Stockify Problem Investing in Stocks is great way to grow money Picking the right stocks for you can get tedious and confusing Too many things to foll

Sanchitha Dinesh 1 Mar 20, 2022
Ezviz IoT Home SDK iOS Demo.

SDK Demo SDK Demo 演示了萤石 App SDK 的接入流程以及 SDK 开放的功能，因此通过参考 Demo 可以基本解决再接入过程中碰到的问题。 Demo 模块介绍： 登录模块：包含注册，登录，忘记密码功能 家庭模块：包含添加家庭，删除家庭，修改家庭信息，邀请家庭成员和移除家庭成员等

萤石.开放平台 1 May 27, 2022
Running Swift automatic differentiation on iOS

Differentiation Demo This is an example of Swift's automatic differentiation running on iOS. It is a modified version of the game from ARHeadsetKit tu

Philip Turner 7 Apr 27, 2022
[yolov5] + [ios] + [tensorflow lite]

YOLOv5 - TensorFlow Lite Object Detection iOS Example Application iOS Versions Supported: iOS 12.0 and above. Xcode Version Required: 10.0 and above O

Inpyo Hong 14 Dec 12, 2022
Resource monitor - A flutter plugin for Android and IOS to monitor CPU and RAM usage of device.

resource_monitor A flutter plugin for Android and IOS to monitor CPU and RAM usage of device. TODO Implement Android Side of this plugin. Add listener

Skandar Munir 1 Nov 11, 2022
Slot-machine - SwiftUI - iOS App Slot Machine game

Slot Machine SwiftUI Masterclass project iPhone, iPad and Mac (with Catalyst framework) Complex interface Extensions and custom View Modifiers Input t

Arthur Neves 1 Sep 12, 2022
This repo contains beginner examples to advanced in swift. Aim to create this for learning native iOS development.

iOS-learning-with-swift-22 This repo contains beginner examples to advanced in swift. Aim to create this for learning native iOS development. Oh, you

Umesh Jangid 0 Jan 9, 2022
Hand-gesture recognition on iOS app using CoreML

GestureAI-CoreML-iOS Hand-gesture recognizer using CoreML Demo Screenshots This app is using RNN(Recurrent Neural network) with CoreML on iOS11. The m

null 151 Nov 2, 2022
Tiny YOLO for iOS implemented using CoreML but also using the new MPS graph API.

YOLO with Core ML and MPSNNGraph This is the source code for my blog post YOLO: Core ML versus MPSNNGraph. YOLO is an object detection network. It can

Matthijs Hollemans 900 Dec 31, 2022
Gyros-identifier - An iOS app capable of detecting whether a photo contains a gyros or not

gyros-identifier ??️ In case you enter a shady restaurant, with suspicious price

Alexandros Tzimas 3 Feb 25, 2022
Pose Estimation on iOS with TensorFlow Lite

This project is Pose Estimation on iOS with TensorFlow Lite. If you are interested in iOS + Machine Learning, visit here you can see various DEMOs. 2D

tucan9389 125 Nov 28, 2022
Spokestack: give your iOS app a voice interface!

Spokestack provides an extensible speech recognition pipeline for the iOS platform. It includes a variety of built-in speech processors for Voice Acti

Spokestack 28 Oct 19, 2022