[开源] Tensorflow CNN RestNet50网络 离线鉴黄识别 WebAPI

前言

参考前两篇文章：

【AI】Tensorflow CNN RestNet50网络 离线鉴黄识别 API

【AI】Tensorflow CNN 离线鉴黄识别 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}

测试截图