/python-nsfw

A Python package for analyzing images using Caffe and Yahoo's open_nsfw models.

Primary LanguagePythonMIT LicenseMIT

python-nsfw

A Python package for analyzing images using Caffe and Yahoo's open_nsfw models.

PyPI Python

Prerequisites

python-nsfw requires Python 3 and runs on Caffe-compatible systems.

Caffe

If you're using macOS and Homebrew, the easiest way to install Caffe is with Homebrew:

brew install caffe

Alternatively, you could use a Docker Linux image (e.g. debian:stretch-slim) and install the necessary dependencies:

apt update && apt install caffe-cpu python3 python3-pip wget

Installation

python-nsfw may be installed using pip:

pip3 install nsfw

Usage

import PIL.Image as Image

from nsfw import classify

image = Image.open("/path/to/image.jpg")
sfw, nsfw = classify(image)

print("SFW Probability: {}".format(sfw))
print("NSFW Probability: {}".format(nsfw))

License

python-nsfw is freely available under the MIT License. Use it, learn from it, fork it, improve it, change it, tailor it to your needs.

python-nsfw ships with copies of Yahoo's open_nsfw Caffe models (nsfw/deploy.prototxt and nsfw/resent_50_1by2_nsfw.caffemodel) which are licensed under the BSD 2-Clause License.