Skip to content
Suitable for Work (NSFW) classification
Python Shell
Branch: master
Clone or download
郑午 郑午
郑午 and 郑午 nsfw
Latest commit 82cfc0b Jan 21, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data/models/1547856517 nsfw Jan 21, 2019
eagle nsfw Jan 21, 2019
resnet nsfw Jan 21, 2019
.gitignore nsfw Jan 21, 2019
LICENSE Initial commit Jan 21, 2019
README.md nsfw Jan 21, 2019
nsfw_predict.py nsfw Jan 21, 2019
requirements.txt nsfw Jan 21, 2019
serving_client.py nsfw Jan 21, 2019
start_tensorflow_serving.sh nsfw Jan 21, 2019

README.md

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

You can’t perform that action at this time.