github地址

https://github.com/mdietrichstein/tensorflow-open_nsfw

模型数据

https://github.com/alex000kim/nsfw_data_scraper

github开源的训练好的模型

https://github.com/rockyzhengwu/nsfw

使用

需要python3环境
执行pip3 install -r requirements.txt安装所需要的库

单个图片鉴黄

例:python3 nsfw_predict.py 10.jpg
屏幕截图 2021-08-08 130204.jpg
输出json数据
屏幕截图 2021-08-08 130250.jpg
class是最有可能的图片类型

类型共有5种:
porn- 色情图片
hentai- 无尽的图像,但也包括色情图纸
sexy-色情图片,但不是色情。想想裸照,花花公子,比基尼等。
neutral-安全的工作中性图像的日常事物和人
drawings-安全的工作图纸(包括动画)

可根据需要,修改py文件,自定义使用。

最后修改:2021 年 08 月 08 日 01 : 06 PM
如果觉得我的文章对你有用,请随意赞赏