🌟前言
参考前两篇文章:
- 🔗【AI】Tensorflow CNN RestNet50网络 离线鉴黄识别 API
- 🔗【AI】Tensorflow CNN 离线鉴黄识别 API
主要是都是打包好的python程序,然后有几个好兄弟说跑不起来,刚好今天有空,把代码发出来(之前的不知道丢哪儿去了)。
Github地址:https://github.com/SwaggyMacro/pynsfw
项目打包地址(模型文件超过100MB Github传不了):
使用说明
首先安装项目依赖,都已经生成好了。
直接在项目根目录执行:
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}
🌃测试截图
