• Stars
    star
    559
  • Rank 79,673 (Top 2 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created almost 6 years ago
  • Updated almost 6 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Not Suitable for Work (NSFW) classification

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