NSFW Model
This repo contains code for running Not Suitable for Work (NSFW) classification.
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
- convert source to tfrecord user
convert_image_to_tfrecord.py
- 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