前言
之前用雅虎的TFLITE模型做过一次二分类的鉴黄（https://b.julym.com/sharetutorial/1155.html），这次是自己训练的模型实现五分类NSFW识别，训练集是从Github上的 nsfw_data_scraper 项目下载下来的（该项目提供了五个分类的图片Url直链，需要自行写脚本下载（项目中提供的脚本因为**原因没办法用...），分别是drawings、hentai、neutral、sexy、porn。）
最后训练出来的模型准确度在90%左右。识别速度大概40MS左右，第一次识别会慢一些（加载库），后面就快了。
示例图
（由于Google Adsense（我的金主爸爸）不能放出裸露 性诱惑 性暗示等图片 所以 我打码了@(脸红)）
使用说明
[info]Windows 系统[/info]
Windows 只需要下载打包好的程序解压出来运行“Main.exe”，然后输入IP以及端口即可（默认为0.0.0.0:8777）
下载链接：https://pan.baidu.com/s/1tGi3jiExNraRtTmOSBbsgw
提取码：6666
[info]Linux 系统[/info]
Linux 需要安装Python后下载提供的Python代码以及安装相应所需库，
然后执行“python flaskWebApi.py”即可。
下载链接：https://pan.baidu.com/s/1WAo7tMdTeMhy4AVYBpXEuQ
提取码：6666
调用说明
请求URL：http://IP:PORT/nsfw (默认是http://127.0.0.1:8777/nsfw)
传递参数：img
参数内容为base64编码过的二进制图片数据
返回内容示例：
{"predict": "sexy", "drawings": "2.2605875e-06", "hentai": "2.7984765e-06", "neutral": "5.642233e-07", "porn": "2.3740056e-06", "sexy": "0.999992", "took": 40}
请求后返回一段JSON，predic 为预测结果，took为消耗时间，其他五项为可能性值。
公用的API接口就不放出来了，上次超星查题的接口被大批量爆破，直接就把服务器挤爆了，直接找我要题库数据库我可能直接给你了，这么搞（还是大批量的用代理IP进行请求）。。现在接口我已经停掉了，估计被他爬了十几万题了，也有可能仅仅只是想进行CC攻击。反正他死了一万个妈！@[鄙视]
下载学习一下，感谢大佬@(doge)
博主你好：要增加训练的模型类型，应该怎么加呢？
自己跑的训练么？定义全连接层的时候改一下分类数就好了，然后标签和训练集都增加一类就行了
自己跑其他类型的训练库，可以简单弄个教程吗？感谢博主。
最近没啥空，抱歉哈。
不过这个网上挺多教程的
博主我用Linux跑您的代码，提示Allocation of 235929600 exceeds 10% of free system memory.
然后就被Killed掉了，怎么解决呢(´；ω；`)
可能 内存不够？可以试试nodejs版的，占用没python那么大，可以参考：https://github.com/infinitered/nsfwjs
好的谢谢
喔，这篇文章是很久以前弄的了，模型文件太大了，效果也不好，后面用resnet50重新训练了一个，参见：https://b.julym.com/original/1238.html
我看看，谢谢大佬Thanks♪(･ω･)ﾉ