|
--- |
|
license: apache-2.0 |
|
base_model: |
|
- google/vit-base-patch16-384 |
|
tags: |
|
- generated_from_trainer |
|
datasets: |
|
- imagefolder |
|
pipeline_tag: image-classification |
|
--- |
|
|
|
<!-- This model card has been generated automatically according to the information the Trainer had access to. You |
|
should probably proofread and complete it, then remove this comment. --> |
|
|
|
# nsfw-image-detector |
|
|
|
This model is a fine-tuned version of [google/vit-base-patch16-384](https://huggingface.co/google/vit-base-patch16-384) on the imagefolder dataset. |
|
It achieves higher accurate than 224 model. |
|
|
|
## Model description |
|
|
|
Labels: ['drawings', 'hentai', 'neutral', 'porn', 'sexy'] |
|
|
|
|
|
## Training and evaluation data |
|
|
|
A custom dataset about 28k images, if you need to improve your domain's accurate, you can contribute the dataset to me. |
