https://github.com/hollance/coreml-training

Source code for my blog post series "On-device training with Core ML"
https://github.com/hollance/coreml-training

coreml edge-computing ios machine-learning on-device-ai swift

Last synced: 11 months ago
JSON representation

Source code for my blog post series "On-device training with Core ML"

• Host: GitHub
• URL: https://github.com/hollance/coreml-training
• Owner: hollance
• License: mit
• Created: 2019-07-19T09:05:24.000Z (over 6 years ago)
• Default Branch: master
• Last Pushed: 2019-12-01T22:14:55.000Z (about 6 years ago)
• Last Synced: 2025-03-20T21:24:06.546Z (11 months ago)
• Topics: coreml, edge-computing, ios, machine-learning, on-device-ai, swift
• Language: Jupyter Notebook
• Size: 9.27 MB
• Stars: 101
• Watchers: 7
• Forks: 11
• Open Issues: 4
• Metadata Files:
  • Readme: README.markdown
  • License: LICENSE

Awesome Lists containing this project

README

          
# Training with Core ML 3

This is the sample code for my blog post series [On-device training with Core ML](https://machinethink.net/blog/coreml-training-part1/).

Included are:

- **Dataset**: a small dataset of 30 training images and 15 test images

- **iOS App**: the source code of the demo app described in the blog post

- **Models**: the empty and pre-trained models used by the app

    - **TuriOriginal.mlmodel**: the SqueezeNet classifier trained by Turi Create

    - **HandsTuri.mlmodel**: the TuriOriginal model but made updatable

    - **HandsEmpty.mlmodel**: like HandsTuri but with a classifier layer that has random weights

    - **HandskNN.mlmodel**: like TuriOriginal but with an untrained k-Nearest Neighbors classifier

- **Scripts**: 

    - **make_nn.py**: converts TuriOriginal.mlmodel to HandsTuri and HandsEmpty.mlmodel

    - **make_knn.py**: creates the k-Nearest Neighbor model, HandskNN.mlmodel

    - **TuriCreate.ipynb**: the notebook used to train TuriOriginal.mlmodel

Credits:

- Camera icon made by [Daniel Bruce](https://www.flaticon.com/authors/daniel-bruce) from [www.flaticon.com](https://www.flaticon.com/) and is licensed by [CC 3.0 BY](http://creativecommons.org/licenses/by/3.0/).

- Picture icon made by [Dave Gandy](https://www.flaticon.com/authors/dave-gandy) from [www.flaticon.com](https://www.flaticon.com/) and is licensed by [CC 3.0 BY](http://creativecommons.org/licenses/by/3.0/).

The source code is copyright 2019 M.I. Hollemans and licensed under the terms of the MIT license.