Downloads · 30 days
45
0% of all-time downloads
vit-base-patch16-224-hentai
carbon225/vit-base-patch16-224-hentai
vit-base-patch16-224-hentai is a image classification model from carbon225. Use it when you need a label for an image. It is set up for transformers. The card lists the license as cc0-1.0.
This is Google's vit-base-patch16-224-in21k finetuned for flagging images according to vndb.org with 3 classes: - safe - suggestive - explicit
All-time downloads
60.3K
Public
Parameters
85.8M
1.4 GB on disk
Likes
22
Public
Repo makeup
Click a slice to open those files.
.bin343 MB · 50%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- cc0-1.0
- Model type
- vit
- Access
- Public
- Created
- Sep 30, 2022
- Updated
- Jul 4, 2023
- SHA
- d072322c
Datasets
Model card
From the Hugging Face model README
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.
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- cc0-1.0
- Created
- Sep 30, 2022
- Updated
- Jul 4, 2023