File size: 799 Bytes
1c80f20 11b3e2d 1c80f20 0e00ce4 1c80f20 11b3e2d 1c80f20 11b3e2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
---
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. |