Need help with nsfw?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

rockyzhengwu
505 Stars 199 Forks Apache License 2.0 2 Commits 7 Opened issues

Description

Not Suitable for Work (NSFW) classification

Services available

!
?

Need anything else?

Contributors list

# 59,234
Python
Shell
1 commit

NSFW Model

This repo contains code for running Not Suitable for Work (NSFW) classification.

online demo

Usage

script

python nsfw_predict.py /tmp/test/test.jpeg

result :

bash
{'class': 'sexy', 'probability': {'drawings': 0.008320281, 'hentai': 0.0011919827, 'neutral': 0.13077603, 'porn': 0.13146976, 'sexy': 0.72824186}}

can find the meaning of every label at repo nsfwdatascrapper

Deploy by TensorFlow Serving

your have to install Tensorflow Serving first

start the server

bash
./start_tensorflow_serving.sh

test server

bash
python serving_client.py /tmp/test/test.jpeg

Train

train code at resnet

train a new model

  1. convert source to tfrecord user
    convert_image_to_tfrecord.py
  2. train a model from scratch or fine tune

the model code copy from Tensorflow offical model

Data

you can find the detail at repo nsfwdatascrapper

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.