Swifter-AI / SentimentVisionDemo

利用神经网络判断评估给定图像的情感色彩

Styles Vision Demo

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

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 StylesVision.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, [e-mail]

Credits

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

References

• Caffe
• Apple Machine Learning
• Vision Framework
• CoreML Framework
• coremltools