opennsfw-standalone 0.0.3

Stand-alone wrapper for Yahoo's OpenNSFW model using Tensorflow.

opennsfw-standalone

A small library for intergrating yahoo/open_nsfw model directly into an application.

This library wraps an Tensorflow version of yahoo/open_nsfw, provided by tensorflow-open_nsfw. The TensorFlow model was exported so there is no direct dependency on yahoo/open_nsfw or tensorflow-open_nsfw.

Prerequisites

• Tensorflow 2.x
• NumPy 1.x

Installation

$ pip install opennsfw-standalone

Usage

Python

import sys

from opennsfw_standalone import OpenNSFWInferenceRunner


inference_runner = OpenNSFWInferenceRunner.load()

for image_filename in sys.argv[1:]:
    with open(image_filename, "rb") as fp:
        nsfw_score = inference_runner.infer(fp.read())
        print(image_filename, nsfw_score)

CLI

$ python -m opennsfw_standalone myimage.jpeg mysecondimage.jpeg
myimage.jpg 0.9997739
mysecondimage.jpg 0.9984438

License

Although this repository is MIT licensed, the model itself is subject to the BSD 2-Clause license Yahoo provided.