Skip to content

cocoa-ai/StylesVisionDemo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Styles Vision Demo

A Demo application using Vision and CoreML frameworks to detect the most likely style of the given image.

StylesVisionDemo

Model

This demo is based on the "Finetuning CaffeNet on Flickr Style" neural network classifier, which was converted from original Caffe model to CoreML model using coremltools python package.

Requirements

  • Xcode 9
  • iOS 11

Installation

git clone https://github.com/cocoa-ai/StylesVisionDemo.git
cd StylesVisionDemo
pod install
open Styles.xcworkspace/

Download the CoreMl model and add the file to "Resources" folder in the project's directory.

Build the project and run it on a simulator or a device with iOS 11.

Conversion

cd Convert
./download.sh
python convert.py

Author

Vadym Markov, [email protected]

Credits

Original "Finetuning CaffeNet on Flickr Style" Caffe model

References