【AI】Tensorflow CNN 离线鉴黄识别 API
前言
之前用雅虎的TFLITE模型做过一次二分类的鉴黄(https://b.julym.com/sharetutorial/1155.html),这次是自己训练的模型实现五分类NSFW识别,训练集是从Github上的 nsfw_data_scraper 项目下载下来的(该项目提供了五个分类的图片Url直链,需要自行写脚本下载(项目中提供的脚本因为**原因没办法用...),分别是drawings、hentai、neutral、sexy、porn。)
最后训练出来的模型准确度在90%左右。识别速度大概40MS左右,第一次识别会慢一些(加载库),后面就快了。
示例图
(由于Google Adsense(我的金主爸爸)不能放出裸露 性诱惑 性暗示等图片 所以 我打码了)
使用说明
Windows 只需要下载打包好的程序解压出来运行“Main.exe”,然后输入IP以及端口即可(默认为0.0.0.0:8777)
下载链接:https://pan.baidu.com/s/1tGi3jiExNraRtTmOSBbsgw
提取码:6666
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攻击。反正他死了一万个妈!
原文链接:【AI】Tensorflow CNN 离线鉴黄识别 API
Macro's Blog 版权所有,转载请注明出处。
下载学习一下,感谢大佬
博主你好:要增加训练的模型类型,应该怎么加呢?
自己跑的训练么?定义全连接层的时候改一下分类数就好了,然后标签和训练集都增加一类就行了
自己跑其他类型的训练库,可以简单弄个教程吗?感谢博主。
最近没啥空,抱歉哈。
不过这个网上挺多教程的
博主我用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♪(・ω・)ノ