Image Aesthetics Estimation (WACV18)
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
arod
demo
training
.gitignore
LICENSE
README.md
teaser.jpg

README.md

Will People like Your Image?

TensorFlow implementation of estimating image aesthetics

Will people like your image
Katharina Schwarz Patrick Wieschollek, Hendrik P.A. Lensch
WACV 2018

Training

The original model was trained on an out-dated TensorFlow training interface. Here, we provide a multi-GPU training script based on TensorPack. For more details, see the training directory.

Video Spotting

We provide an example to do video spotting. The steps are:

user@host $ VIDEO=original.mp4
user@host $ avconv -i original.mp4 -s 256x256 -strict -2 echo ${VIDEO%.mp4}-small.mp4  # resize the video to match the network input
user@host $ python apply.py --mp4 ${VIDEO%.mp4}-small.mp4 # apply our neural network
user@host $ python gui.py # use the GUI to display the video and the scores

Click Load Video and select the original high-res video.

Training Data

The file arod/list.txt provides the url to images used during training. The format is url;faves;views.