metadata
license: cc0-1.0
widget:
- src: >-
https://huggingface.co/carbon225/vit-base-patch16-224-hentai/resolve/main/samples/1.jpeg
- src: >-
https://huggingface.co/carbon225/vit-base-patch16-224-hentai/resolve/main/samples/2.jpeg
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).
Because the dataset contains questionable images, I will not publish it.
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.