Skip to content

Latest commit

 

History

History
39 lines (26 loc) · 1.09 KB

README.md

File metadata and controls

39 lines (26 loc) · 1.09 KB

Deep Neural Network Artwork Classifier

This is an artwork classifier, which classifies paintings of more than 200 artists such as Claude Monet and Pablo Picasso. A running demo is deployed at Heroku.

The classifier is based on a deep neural network (DNN) implemented and trained with tensorflow. The model architecture is constituted by a multi-layer convolutional neural network, which is followed by a fully connected layer and completed with a softmax readout layer.

Training was done based on a cross-entropy error function.

Install Dependencies

$ virtualenv -p python2.7 venv
$ source venv/bin/activate
$ pip install -r requirements.txt

Start Web Server

$ gunicorn main:app --log-file=- --timeout=600 --preload

Train Model

$ python preprocess.py
$ python train.py