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
- 38
Model tree for LukeJacob2023/nsfw-image-detector-384
Base model
google/vit-base-patch16-384