nicolas-raoul / selfie-or-not

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

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

License:Apache License 2.0


Languages

Language:Python 81.2%Language:Shell 18.8%