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

4,511 views次阅读
12条评论

📮广告: 免费体验ChatGPT

免费体验最强ChatGPT: https://chat.ncii.cn/
注册即送100积分。
免费体验最强ChatGPT:https://chat.ncii.cn/

前言

之前用雅虎的TFLITE模型做过一次二分类的鉴黄（https://b.julym.com/sharetutorial/1155.html），这次是自己训练的模型实现五分类NSFW识别，训练集是从Github上的 nsfw_data_scraper 项目下载下来的（该项目提供了五个分类的图片Url直链，需要自行写脚本下载（项目中提供的脚本因为**原因没办法用...），分别是drawings、hentai、neutral、sexy、porn。）

最后训练出来的模型准确度在90%左右。识别速度大概40MS左右，第一次识别会慢一些（加载库），后面就快了。

示例图

 

【AI】Tensorflow

【AI】Tensorflow

（由于Google Adsense（我的金主爸爸）不能放出裸露 性诱惑 性暗示等图片 所以 我打码了@(脸红)）

使用说明

[info]Windows 系统[/info]

Windows 只需要下载打包好的程序解压出来运行“Main.exe”，然后输入IP以及端口即可（默认为0.0.0.0:8777）

下载链接：https://pan.baidu.com/s/1tGi3jiExNraRtTmOSBbsgw
提取码：6666

[info]Linux 系统[/info]

Linux  需要安装Python后下载提供的Python代码以及安装相应所需库，

然后执行“python flaskWebApi.py”即可。

下载链接：https://pan.baidu.com/s/1WAo7tMdTeMhy4AVYBpXEuQ
提取码：6666

调用说明

请求URL：http://IP:PORT/nsfw (默认是http://127.0.0.1:8777/nsfw)

传递参数：img

参数内容为base64编码过的二进制图片数据

返回内容示例：

{"predict": "sexy", "drawings": "2.2605875e-06", "hentai": "2.7984765e-06", "neutral": "5.642233e-07", "porn": "2.3740056e-06", "sexy": "0.999992", "took": 40}

请求后返回一段JSON，predic 为预测结果，took为消耗时间，其他五项为可能性值。

 

公用的API接口就不放出来了，上次超星查题的接口被大批量爆破，直接就把服务器挤爆了，直接找我要题库数据库我可能直接给你了，这么搞（还是大批量的用代理IP进行请求）。。现在接口我已经停掉了，估计被他爬了十几万题了，也有可能仅仅只是想进行CC攻击。反正他死了一万个妈！@[鄙视]

2021-01-25
赏
Swaggy Macro
版权声明：本站原创文章，由 Swaggy Macro 2021-01-25发表，共计1054字。
转载说明：除特殊说明外本站文章皆由CC-4.0协议发布，转载请注明出处。
【原创·开源】易语言RockStarID查询工具源代码
【AI】Tensorflow CNN RestNet50网络 离线鉴黄识别 API
“少女是造物者的馈赠✨ “
【Linux】Ubuntu 配置V2ray 加速Git HTTP配置PAC 国外国内分流
评论（12条评论）
验证码
красный 评论达人 LV.1
2021-01-28 09:58:32 回复

下载学习一下，感谢大佬@(doge)

WindowsWindowsEdgeEdge88.0.705.50
chen 评论达人 LV.1
2021-11-29 17:43:48 回复

博主你好：要增加训练的模型类型，应该怎么加呢？

WindowsWindowsChromeChrome96.0.4664.45
    Macro 博主
    2021-11-29 21:43:30 回复

    @chen 自己跑的训练么？定义全连接层的时候改一下分类数就好了，然后标签和训练集都增加一类就行了

    WindowsWindowsEdgeEdge96.0.1054.34
      chen 评论达人 LV.1
      2021-11-29 22:55:02 回复

      @Macro 自己跑其他类型的训练库，可以简单弄个教程吗？感谢博主。

      WindowsWindowsChromeChrome95.0.4638.54
        Macro 博主
        2021-11-30 19:48:17 回复

        @chen 最近没啥空，抱歉哈。
        不过这个网上挺多教程的

        WindowsWindowsEdgeEdge96.0.1054.34
aoguai 评论达人 LV.1
2022-02-28 20:16:27 回复

博主我用Linux跑您的代码，提示Allocation of 235929600 exceeds 10% of free system memory.
然后就被Killed掉了，怎么解决呢(´；ω；`)

WindowsWindowsEdgeEdge98.0.1108.50
    Macro 博主
    2022-02-28 22:15:00 回复

    @aoguai 可能 内存不够？可以试试nodejs版的，占用没python那么大，可以参考：https://github.com/infinitered/nsfwjs

    WindowsWindowsChromeChrome98.0.4758.102
      aoguai 评论达人 LV.1
      2022-02-28 22:19:52 回复

      @Macro 好的谢谢

      WindowsWindowsEdgeEdge98.0.1108.50
        Macro 博主
        2022-02-28 22:21:35 回复

        @aoguai 喔，这篇文章是很久以前弄的了，模型文件太大了，效果也不好，后面用resnet50重新训练了一个，参见：https://b.julym.com/original/1238.html

        WindowsWindowsChromeChrome98.0.4758.102
          aoguai 评论达人 LV.1
          2022-02-28 22:23:45 回复

          @Macro 我看看，谢谢大佬Thanks♪(･ω･)ﾉ

          WindowsWindowsEdgeEdge98.0.1108.50