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.
- Downloads last month
- 11,781