English | 中文版

NSFW

NSFW - not safe for work

Description

Trained on 600,000 labled pictures:

• porn - pornography images
• hentai - hentai images, but also includes pornographic drawings
• sexy - sexually explicit images, but not pornography. Think nude photos, playboy, bikini, etc.
• neutral - safe for work neutral images of everyday things and people
• drawings - safe for work drawings (including anime)

Requeriments

pytorch 1.0+
If you want pytorch 0.4, please download V1 release.

Usage

#train
python train.py --model resnet101 --epochs 90 --batch-size 512 --checkpoint ./checkpoint --data-dir ./data

#test
python test_confusion_matrix.py

#predict
python predict --model resnet101 --checkpoint ./checkpoint/x

#if your machine has connected to the internet and you dosen't want to download the image to your disk
cat urls.txt | python predict_url.py

Training data source

Special thanks to the nsfw_data_scraper for the training data. If you're interested in a more detailed analysis of types of NSFW images, you could probably use this repo code with this data.
If you want make better result.Contact me.I can provide you the best training data.

Current status

Sexy and porn is a little similar.In my view,it does'nt matter.  
 

SEXY

   

NETURAL

Detail

I have tried various methods include some pretrained models like resnet/inceptionv3 and data augumentation and finetuing.

Here are some tips which make a greate effect to the final result:

• Make batch size bigger.(the bigger the better since I make it 512 with my p40)
• Use pretrained model.(you can use torchvision. pretrained model can help your model convergence more faster)
• Lock some layer and finetune FC.(after train_init.py then lock some layer just finetune the FC)
• Adjust learning rate.(make lr dynamic when training in order to get saddle point)
• Select appropriate pretrained model.(I choose resnet101 since it receive better result than resnet50 or inceptionv3)

Thanks

Thanks for my wife FeiFei Li. She gave me lots of encouragement. And made the beautiful logo for NSFW preject.
Thanks for my workmate Kuai Li. He gave me lots of good suggestion.

Join us

If you have good points.Join us!

References

https://github.com/GantMan/nsfw_model