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.

Downloads last month
65
Safetensors
Model size
86.1M params
Tensor type
F32
·
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.

Model tree for LukeJacob2023/nsfw-image-detector-384

Finetuned
(22)
this model