Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/hollance/CoreMLHelpers

Types and functions that make it a little easier to work with Core ML in Swift.
https://github.com/hollance/CoreMLHelpers

core-ml coreml ios machine-learning swift

Last synced: about 2 months ago
JSON representation

Types and functions that make it a little easier to work with Core ML in Swift.

Host: GitHub
URL: https://github.com/hollance/CoreMLHelpers
Owner: hollance
License: mit
Created: 2017-08-05T21:27:08.000Z (over 7 years ago)
Default Branch: master
Last Pushed: 2022-09-06T03:05:25.000Z (over 2 years ago)
Last Synced: 2024-07-03T02:50:53.794Z (6 months ago)
Topics: core-ml, coreml, ios, machine-learning, swift
Language: Swift
Size: 952 KB
Stars: 1,340
Watchers: 40
Forks: 186
Open Issues: 12
Metadata Files:
- Readme: README.markdown
- License: LICENSE.txt

Awesome Lists containing this project

awesome-ios - CoreMLHelpers

README

# CoreMLHelpers

This is a collection of types and functions that make it a little easier to work with Core ML in Swift.

Some of the things CoreMLHelpers has to offer:

- [convert images to `CVPixelBuffer` objects and back](Docs/CVPixelBuffer.markdown)
- [`MLMultiArray` to image conversion](Docs/MultiArray2Image.markdown)
- [handy functions](Docs/HandyFunctions.markdown) to get top-5 predictions, argmax, and so on
- [non-maximum suppression](Docs/NMS.markdown) for bounding boxes

Experimental features:

- [a more Swift-friendly version of `MLMultiArray`](Docs/SwiftyMultiArray.markdown)

Let me know if there's anything else you'd like to see added to this library!

## :bulb: Tip: Get the Core ML Survival Guide

If Core ML is giving you trouble --- or if you want to learn more about using the Core ML and Vision APIs --- then check out my book [Core ML Survival Guide](https://leanpub.com/coreml-survival-guide). It has 400+ pages of Core ML tips and tricks.

I wrote the [Core ML Survival Guide](https://leanpub.com/coreml-survival-guide) because the same questions kept coming up on Stack Overflow, on the Apple Developer Forums, and on this GitHub repo. Core ML may appear easy-to-use at first --- but if you want to go beyond the basics, the learning curve suddenly becomes very steep. My goal with this book is to make the advanced features of Core ML accessible to everyone too.

The [Core ML Survival Guide](https://leanpub.com/coreml-survival-guide) currently has over 80 chapters and includes pretty much everything I know about Core ML. As I learn new things I'll keep updating the book, so you'll always have access to the most up-to-date knowledge about Core ML. Cheers!

## How to use CoreMLHelpers

Copy the source files from the **CoreMLHelpers** folder into your project. You probably don't need all of them, so just pick the files you require and ignore the rest.

> **Note:** A lot of the code in CoreMLHelpers is only intended as a demonstration of how to approach a certain problem. There's often more than one way to do it. It's quite likely you will need to customize the code for your particular situation, so consider these routines a starting point.

### Why can't I use this with Swift Package Manager / Carthage / CocoaPods?

I believe a proper framework should have a well-thought-out API but CoreMLHelpers is a hodgepodge of helper functions that isn't particularly well-organized. Putting this into a package makes things more complicated than necessary. Just copy the one or two source files you need into your project, and adapt them to your needs.

## Read more about Core ML

- [Apple's Machine Learning page](https://developer.apple.com/machine-learning/)
- [My blog: machinethink.net](https://machinethink.net/blog)

## TODO

- proper unit tests
- add more numpy-like functionality to `MultiArray` (and fix the bugs!)

## License