metadata
license: apache-2.0
base_model:
- google/vit-base-patch16-384
tags:
- generated_from_trainer
datasets:
- imagefolder
pipeline_tag: image-classification
nsfw-image-detector
This model is a fine-tuned version of 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.