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
4,141
Safetensors
Model size
86.1M params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

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

Finetuned
(22)
this model