nsfw_model

收录理由

这个Keras模型为需要做内容审核的团队提供了一个可落地的起点，尤其是那些允许用户上传图片或聊天附件的平台。它能把图像归入五个类别：绘画、hentai、中性、色情和性感，省去了自己标注海量样本的功夫。仓库里已经准备好了Inception V3和MobileNet两种预训练权重，你可以通过一小段Python脚本或者自带的命令行工具来处理单张图片、整个文件夹或者批量文件。值得一提的是，这个模型还支撑了更知名的NSFW JS浏览器版本，说明这些权重经受过真实线上用户的检验。如果团队想在针对自身受众微调之前，先拿一个文档齐全的基线来跑通流程，这个项目会是不错的选择。