加入 Gitee

与超过 1200万开发者一起发现、参与优秀开源项目，私有仓库也完全免费：）

此仓库是为了提升国内下载速度的镜像仓库，每日同步一次。原始仓库： https://github.com/rockyzhengwu/nsfw

克隆/下载

README.md 1.14 KB

# NSFW Model

This repo contains code for running Not Suitable for Work (NSFW) classification.

[online demo](http://ai.midday.me/)

## Usage

#### script

```bash
python nsfw_predict.py /tmp/test/test.jpeg
```

result : 
```bash
{'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](https://github.com/alexkimxyz/nsfw_data_scrapper)

#### Deploy by TensorFlow Serving

your have to install [Tensorflow Serving](https://www.tensorflow.org/serving/) first

start the server
```bash
./start_tensorflow_serving.sh
```

test server
```bash
python serving_client.py /tmp/test/test.jpeg
```

## Train

train code at [resnet](./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](https://github.com/tensorflow/models/tree/master/official/resnet)

## Data

you can find the detail at repo [nsfw_data_scrapper](https://github.com/alexkimxyz/nsfw_data_scrapper)

一键复制原始数据按行查看历史

提交于 2019-01-21 11:03 . nsfw

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

Python

1

https://gitee.com/mirrors/nsfw.git

git@gitee.com:mirrors/nsfw.git

mirrors

nsfw

nsfw

master