[开源] 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}

测试截图

评论

  1. aoguai
    5年前
    2022-3-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}
    不过还是谢谢大佬的代码!@(害羞)

    • 博主
      aoguai
      5年前
      2022-3-16 19:20:33

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

  2. 猎隼丶止戈
    5年前
    2022-3-22 10:36:27

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

    https://imgtu.com/i/qKJ81K

    • 博主
      猎隼丶止戈
      5年前
      2022-3-22 10:39:04

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

    • 博主
      猎隼丶止戈
      5年前
      2022-3-22 10:51:54

      colab不好传数据集

发送评论 编辑评论


				
|´・ω・)ノ
ヾ(≧∇≦*)ゝ
(☆ω☆)
(╯‵□′)╯︵┴─┴
 ̄﹃ ̄
(/ω\)
∠( ᐛ 」∠)_
(๑•̀ㅁ•́ฅ)
→_→
୧(๑•̀⌄•́๑)૭
٩(ˊᗜˋ*)و
(ノ°ο°)ノ
(´இ皿இ`)
⌇●﹏●⌇
(ฅ´ω`ฅ)
(╯°A°)╯︵○○○
φ( ̄∇ ̄o)
ヾ(´・ ・`。)ノ"
( ง ᵒ̌皿ᵒ̌)ง⁼³₌₃
(ó﹏ò。)
Σ(っ °Д °;)っ
( ,,´・ω・)ノ"(´っω・`。)
╮(╯▽╰)╭
o(*////▽////*)q
>﹏<
( ๑´•ω•) "(ㆆᴗㆆ)
😂
😀
😅
😊
🙂
🙃
😌
😍
😘
😜
😝
😏
😒
🙄
😳
😡
😔
😫
😱
😭
💩
👻
🙌
🖕
👍
👫
👬
👭
🌚
🌝
🙈
💊
😶
🙏
🍦
🍉
😣
Source: github.com/k4yt3x/flowerhd
颜文字
Emoji
小恐龙
花!
上一篇
下一篇