---
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.