[开源] Tensorflow CNN RestNet50网络 离线鉴黄识别 WebAPI
前言
参考前两篇文章:
【AI】Tensorflow CNN RestNet50网络 离线鉴黄识别 API
主要是都是打包好的python程序,然后有几个好兄弟说跑不起来,刚好今天有空,把代码发出来(之前的不知道丢哪儿去了)。
Github地址:https://github.com/SwaggyMacro/pynsfw
项目打包地址(模型文件超过100MB Github传不了):
百度网盘:https://pan.baidu.com/s/1IsRJnaRWJz-rvrK-dam0cw?pwd=tdbh
阿里云盘:https://www.aliyundrive.com/s/MyfM6cgsbo7
使用说明
首先安装项目依赖,都已经生成好了。
直接在项目根目录执行:
pip install -r requirements.txt
nsfw是整个实现NSFW检测的类,简单调用检测图片方法:
nsfw = Nsfw("h5模型文件路径") print (nsfw.check("图片路径或Image.open后的返回数据")) # 返回示例 # {'ret': 'neutral', 'drawings': '0.0003323109', 'hentai': '6.302585e-06', 'neutral': '0.97176874', 'porn': '0.00476053', # 'sexy': '0.02313208', 'time': 140}
运行"WebApi.py"文件会通过Flask构建一个WebAPI出来,地址是0.0.0.0:54321,端口号和地址可以自己更改。
API调用方式:直接发起Http请求,唯一参数img,参数内容就是base64后的图片数据(不要开头的文件类型"data:image/jpg;base64,")
返回结果示例(ret就是几率最高的分类名称):
{"ret": "neutral", "drawings": "0.0003323109", "hentai": "6.302585e-06", "neutral": "0.97176874", "porn": "0.00476053", "sexy": "0.02313208", "time": 140}
测试截图

原文链接:[开源] Tensorflow CNN RestNet50网络 离线鉴黄识别 WebAPI
Macro's Blog 版权所有,转载请注明出处。
感谢大佬,已经成功运行了。
试了张熊猫头也判断sexy
放在QQ机器人上用上了。
用的大佬的模型,真人效果感觉还行。卡通动漫图片就不行了
{‘ret’: ‘sexy’, ‘drawings’: ‘0.0006150138’, ‘hentai’: ‘0.00012748287’, ‘neutral’: ‘0.0036919876’, ‘porn’: ‘0.010799796’, ‘sexy’: ‘0.98476577’, ‘time’: 1820}
不过还是谢谢大佬的代码!
卡通的训练集少一点,真人的比较多
艾薇封面识别貌似不太行呀….
https://imgtu.com/i/qKJ81K
数据集里这种类似图片太少,都是一些纯porn sexy图片,后面买新显卡了可以再增大数据集训练一次
colab不好传数据集