/nudity

Python nudity detection with re-trained Tensorflow MobileNet Model. Accuracy is 92.2% based on my dataset

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

nudity

Build status

About

Nudity detection with re-trained Tensorflow MobileNet Model. Accuracy is 92.2% based on my dataset.

Installation

$ pip install nudity

Requirements

  • Python3.5+

Usage

via command-line

$ nudity --image=IMAGE_FILE

via Python Module

from nudity import Nudity
nudity = Nudity();
print(nudity.has('/file/path/example.jpg'))
# gives you True or False

print(nudity.score('/file/path/example.jpg'))
# gives you nudity score 0 - 1