Skip to content

rockyzhengwu/nsfw

Repository files navigation

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 :

{'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 nsfw_data_scrapper

Deploy by TensorFlow Serving

your have to install Tensorflow Serving first

start the server

./start_tensorflow_serving.sh

test server

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 nsfw_data_scrapper

About

Not Suitable for Work (NSFW) classification

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published