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

4,040 views次阅读
12条评论

前言

之前用雅虎的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攻击。反正他死了一万个妈!@[鄙视]

Swaggy Macro
版权声明:本站原创文章,由 Swaggy Macro 2021-01-25发表,共计1054字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(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