NudeNet: An ensemble of Neural Nets for Nudity Detection and Censoring

Part 2: Exposed part detection and censoring.

The Why: A major draw back of image classification is not having fine grained control. If we want to blur the exposed parts/ have fine -grained control of which type of images we want to allow, we need Object Detection. This is a first of it’s kind open-source project, which I hope will be helpful to the community.

Total data available for training

Evaluating the model:

Precision and Recall of NudeNet’s Detector
Censored image, using NudeDetector.
# installing the project
pip install git+https://github.com/bedapudi6788/NudeNet
# Using the classifier
from NudeNet import NudeClassifier
classifier = NudeClassifier('classifier_checkpoint_path')
classifier.classify('path_to_nude_image')

--

--

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store