https://github.com/hollance/coreml-survival-guide

Source code for the book Core ML Survival Guide
https://github.com/hollance/coreml-survival-guide

coreml deep-learning ios machine-learning macos swift

Last synced: 15 days ago
JSON representation

Source code for the book Core ML Survival Guide

• Host: GitHub
• URL: https://github.com/hollance/coreml-survival-guide
• Owner: hollance
• License: mit
• Created: 2018-11-16T10:46:08.000Z (almost 6 years ago)
• Default Branch: master
• Last Pushed: 2020-10-09T15:55:29.000Z (about 4 years ago)
• Last Synced: 2023-11-07T18:21:36.692Z (about 1 year ago)
• Topics: coreml, deep-learning, ios, machine-learning, macos, swift
• Language: Python
• Size: 9.83 MB
• Stars: 233
• Watchers: 11
• Forks: 34
• Open Issues: 5
• Metadata Files:
  • Readme: README.markdown
  • License: LICENSE

Awesome Lists containing this project

README

        
# Core ML Survival Guide

This is the source code for the book [Core ML Survival Guide](http://leanpub.com/coreml-survival-guide) -- now updated for iOS 14 and macOS 11.0.

## The book

If you don't have the book yet, here's my sales pitch: The [Core ML Survival Guide](http://leanpub.com/coreml-survival-guide) is more than 80 chapters and almost 500 pages of Core ML tips and tricks. 

Learn the best ways to convert your models to Core ML, how the mlmodel file format works, how to perform model surgery to fix issues with your mlmodel files, how use `MLMultiArray`, and much more!

If you're serious about Core ML, [you need this book](http://leanpub.com/coreml-survival-guide). Seriously! :smile:

## The source code

Included in this source code repo is the following:

**CheckInputImage:** Demo app for iOS that shows how to use a very basic image-to-image model to verify how Vision and the neural network preprocessing options modify the input image.

**MobileNetV2+SSDLite:** Conversion script and demo app for the SSDLite object detection model. (The original model is not included, you'll need to download this first. See the link in **ssdlite.py**.)

**NeuralNetworkBuilder:** Demo of how to write your own Core ML converter. Includes the trained Caffe model.

Scripts:

- **convert_multiarrays_to_floats.py:** Changes the data type of any MultiArray inputs and outputs from `DOUBLE` to `FLOAT32`.

- **convert_to_float16.py:** Changes the weights of the model from 32-bit to 16-bit floating point, cutting the model size in half.

- **deeplab.py:** Demonstration of how to clean up a model by renaming inputs and outputs, removing unnecessary layers, inserting new layers, etc.

- **helpers.py:** Useful utility functions for model surgery.

- **quantize.py:** For quantizing and dequantizing models.

Also check out my other repo [CoreMLHelpers](https://github.com/hollance/CoreMLHelpers) for a collection of helper code that makes it a little easier to work with Core ML in Swift.