HyperNSFW-Caffe

Introduction

This is a Caffe implementation of Not Suitable for Work (NSFW) classification deep neural network. This model is now slightly more accurate than Yahoo's Open NSFW. As a bonus, this model executes ~4x faster. We can run classification with this model at 80msec on the mobile phone. For details, please read the our blog.

Evaluate Models with a single image

Evaluate HyperNSFW:

python eval_image.py --proto model/mobilenet_v2_deploy.prototxt --model model/HyperNSFW.caffemodel --image ./test.jpg

Expected Outputs:

0.9302 - 'n16000001 sfw'
0.0698 - 'n16000002 nsfw'

Finetuning on your own data

Prepare your own data and use related files in the train folder to Finetune on your own data.

Related Projects

HyperNSFW is trained with the following projects, we recommend you to take a look:

• MobileNet-Caffe shicai/MobileNet-Caffe

To Do

• We will release Android, iOS, Linux Projects soon.