A Python package for analyzing images using Caffe and Yahoo's open_nsfw models.
Project description
python-nsfw
A Python package for analyzing images using Caffe and Yahoo's open_nsfw models.
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.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.