7 Star 24 Fork 11

Gitee 极速下载 / nsfw

Create your Gitee Account
Explore and code with more than 6 million developers,Free private repositories !:)
Sign up
此仓库是为了提升国内下载速度的镜像仓库,每日同步一次。 原始仓库: https://github.com/rockyzhengwu/nsfw
Clone or download
Cancel
Notice: Creating folder will generate an empty file .keep, because not support in Git
Loading...
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

Comments ( 1 )

Sign in for post a comment

About

黄图鉴别工具 写在前面:不要找我要训练数据,我是遵纪守法的好公民,训练数据已经删除 spread retract
Cancel

Releases

No release

Contributors

All

Activities

load more
can not load any more
Python
1
https://gitee.com/mirrors/nsfw.git
git@gitee.com:mirrors/nsfw.git
mirrors
nsfw
nsfw
master

Search