English | 中文版
NSFW - not safe for work
Trained on 600,000 labled pictures:
porn
- pornography imageshentai
- hentai images, but also includes pornographic drawingssexy
- sexually explicit images, but not pornography. Think nude photos, playboy, bikini, etc.neutral
- safe for work neutral images of everyday things and peopledrawings
- safe for work drawings (including anime)pytorch 1.0+
If you want pytorch 0.4, please download V1 release.
#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
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.
Sexy and porn is a little similar.In my view,it does'nt matter.
SEXY
NETURAL
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:
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.
If you have good points.Join us!