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

1,292 views次阅读
5条评论

前言

参考前两篇文章:

【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}

测试截图

[开源]

Swaggy Macro
版权声明:本站原创文章,由 Swaggy Macro 2022-03-16发表,共计953字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(5条评论)
验证码
载入中...
aoguai 评论达人 LV.1
2022-03-16 19:14:53 回复

感谢大佬,已经成功运行了。
放在QQ机器人上用上了。
用的大佬的模型,真人效果感觉还行。卡通动漫图片就不行了
@(脸红)试了张熊猫头也判断sexy
{‘ret’: ‘sexy’, ‘drawings’: ‘0.0006150138’, ‘hentai’: ‘0.00012748287’, ‘neutral’: ‘0.0036919876’, ‘porn’: ‘0.010799796’, ‘sexy’: ‘0.98476577’, ‘time’: 1820}
不过还是谢谢大佬的代码!@(害羞)

WindowsWindowsEdgeEdge99.0.1150.39
    Macro 博主
    2022-03-16 19:20:33 回复

    @aoguai 卡通的训练集少一点,真人的比较多

    WindowsWindowsChromeChrome99.0.4844.51
猎隼丶止戈 评论达人 LV.1
2022-03-22 10:36:27 回复

艾薇封面识别貌似不太行呀….

https://imgtu.com/i/qKJ81K

WindowsWindowsChromeChrome98.0.4758.82
    Macro 博主
    2022-03-22 10:39:04 回复

    @猎隼丶止戈 数据集里这种类似图片太少,都是一些纯porn sexy图片,后面买新显卡了可以再增大数据集训练一次

    WindowsWindowsChromeChrome99.0.4844.51
    Macro 博主
    2022-03-22 10:51:54 回复

    @猎隼丶止戈 colab不好传数据集

    WindowsWindowsChromeChrome99.0.4844.51