nicolas-raoul/selfie-or-not

Using machine learning, tell whether an image is a selfie or not

Goal

This project's goal is to detect selfies among random pictures.

Usage

Install the necessary packages, for instance on Ubuntu 2018.04:

sudo apt-get update && apt-get install -y python-numpy python-virtualenv
virtualenv tensorflow_env
pip install tensorflow keras

Then just run:

python run-preprocessed.py

Data samples

All images are from Wikimedia Commons under the same name. You can also use your own images and generate their thumbnails with the included preprocess-images.sh.

selfie-or-not's Issues

Make Android library

Currently the neural network parameters are not even outputted.
They should be saved using tf.saved_model.builder.SavedModelBuilder and an Android library should be developed around that calling SavedModelBundle.load.
The biggest challenge is to reduce the size of the TensorFlow dependencies from 80 MB to as small as possible. This probably involves a lot of deep-level TensorFlow hacking, and thus a good understanding of neural networks.