Giter VIP home page Giter VIP logo

awesome-coreml-models's Introduction

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

  • Exermote - Predicts the exercise, when iPhone is worn on right upper arm. Download | Demo | Reference
  • GestureAI - Recommend an artist based on given location and genre. Download | Demo | Reference
  • Artists Recommendation - Recommend an artist based on given location and genre. Download | Demo | Reference
  • ChordSuggester - Predicts the most likely next chord based on the entered Chord Progression. Download | Demo | Reference

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.

awesome-coreml-models's People

Contributors

carlosmbe avatar chrisdunaetz avatar gpoussel avatar imxieyi avatar julien-c avatar lesyk avatar likedan avatar menant avatar narner avatar novinfard avatar plantvsbirds avatar spekulatius avatar tucan9389 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

awesome-coreml-models's Issues

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

  • Only one item is in this issue
  • The model info contains all the required fields
  • The demo project is compilable
  • Has proper reference
  • 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)

Validate Links

Hello, I wrote a tool that can validate links in a README (or any file). It can be run when someone submits a pull request or pushes a commit to Awesome-CoreML-Models.

This tool is currently being used by

More examples

If you are interested, connect this repo to https://travis-ci.org/ and add a .travis.yml file to the project (you can also use CircleCI or other CI tools).

See https://github.com/dkhamsing/awesome_bot for options, more information ๐Ÿ˜„

"TextRecognition" is not a CoreML example

TextRecognition - Recognizing text using ML Kit built-in model in real-time.

Links to project that uses ML Kit (Google Firebase) and not CoreML (Apple) framework and model.

New Model:NIMA

License

MIT

Summary

Automatically learned quality assessment for images has recently become a hot topic due to its usefulness in a wide variety of applications such as evaluating image capture pipelines, storage techniques and sharing media. Despite the subjective nature of this problem, most existing methods only predict the mean opinion score provided by datasets such as AVA [1] and TID2013 [2]. Our approach differs from others in that we predict the distribution of human opinion scores using a convolutional neural network. Our architecture also has the advantage of being significantly simpler than other methods with comparable performance. Our proposed approach relies on the success (and retraining) of proven, state-of-the-art deep object recognition networks. Our resulting network can be used to not only score images reliably and with high correlation to human perception, but also to assist with adaptation and optimization of photo editing/enhancement algorithms in a photographic pipeline. All this is done without need for a "golden" reference image, consequently allowing for single-image, semantic- and perceptually-aware, no-reference quality assessment.

Model URL

Model Link

Demo URL

PhotoAssessment

Samples

Input: Image[http://i1.hdslb.com/bfs/archive/9892e6f032425fc9e9831fa1ed855318c12702ad.jpg], Output: Double 10 vector [1.614268398952845e-06,0.0006238522473722696,0.02812075242400169,0.2070262581110001,0.3266536593437195,0.332121878862381,0.09285952150821686,0.01259173825383186,2.718873588491988e-07,4.697789393048879e-07]

Checklist

  • Only one item is in this issue
  • The model info contains all the required fields
  • The demo project is compilable
  • Has proper reference
  • If this model takes image as input, an image type is selected instead of multiarray

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

  • Only one item is in this issue
  • The model info contains all the required fields
  • The demo project is compilable (AFTER INCLUDE THE MLMODELS)
  • [?] Has proper reference
  • 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)

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

  • Only one item is in this issue
  • The model info contains all the required fields
  • The demo project is compilable
  • Has proper reference

Size of CoreML model

Hi, I am trying to use CNNEmotions, but his size is too big.

Would it be possible to reduce its size?

Thanks!

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

  • Only one item is in this issue
  • The model info contains all the required fields
  • The demo project is compilable
  • Has proper reference
  • 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)

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

  • Only one item is in this issue
  • The model info contains all the required fields
  • The demo project is compilable
  • Has proper reference
  • 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)

Cool Start Up

This is not an issue I just wanted to make sure this repo had seen this cool start up:

https://lobe.ai/examples

They claim to have lots of CoreML Models from that example page. No way to download from what I saw though.

Regardless, I signed up for their beta.

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 :)

Models for exercising

Do you have any mlmodels using activity classification for exercising? Like push ups and jumping jacks?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.