Skip to content

cocoa-ai/SentimentVisionDemo

Repository files navigation

Sentiment Vision Demo

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

SentimentVisionDemo

Model

This demo is based on the "Fine-tuning CNNs for Visual Sentiment Prediction" 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-ml/SentimentVisionDemo.git
cd SentimentVisionDemo
pod install
open SentimentVision.xcworkspace/

Download the CoreMl model and drag the file into your project.

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

From Pixels to Sentiment: Fine-tuning CNNs for Visual Sentiment Prediction

References