/PlacesClassifierCoreML

CoreML models and sample code for classifying places based on the Places dataset http://places.csail.mit.edu/index.html

Primary LanguageObjective-C

Places classifier with Core ML

This repo contains sample code and trained models for classifying places with a neural network on the Places dataset http://places.csail.mit.edu/index.html.

This project was originally forked from https://github.com/chenyi1989/CoreMLDemo. Their original readme is below

CoreMLDemo

A demo for iOS machine learning framework : Core ML

Only Xcode9 and above are supported.

Model Places205-GoogLeNet comes from [Apple Machine Learning] (https://developer.apple.com/machine-learning/).

See more details about this demo in Chinese.

If an error like GoogLeNetPlaces.h not found occurs when you run this demo, try to drag the mlmodel file into the project and check the option in Target Membership on the right. It seems like a bug of Xcode 9 beta.

References

Getting a Core ML Model

Integrating a Core ML Model into Your App

How to convert UIImage to CVPixelBufferRef[Chinese]