ViT for NSFW classification

Model info

This is Google's vit-base-patch16-224-in21k finetuned for flagging images according to vndb.org with 3 classes:

• safe
• suggestive
• explicit

Training data

The model was trained on the vndb.org database dump using full size screenshots ( sf in the database dump).The dataset can be loaded from carbon225/vndb_img .

Intended use

The model can be used for flagging anime-style images for sexual content.It can also be finetuned on other tasks related to anime images.

ViT用于NSFW分类

模型信息

这是Google的 vit-base-patch16-224-in21k ，根据 vndb.org 对图像进行了微调，具有3个类别：

• safe
• suggestive
• explicit

训练数据

该模型是在vndb.org database dump 上使用全尺寸的屏幕截图（数据库转储中的sf）进行训练的。数据集可以从 carbon225/vndb_img 加载。

预期用途

该模型可用于标记动画风格的图像中的性内容。它还可以在与动画图像相关的其他任务上进行微调。