MMLForge Search models, datasets, tasks… Start free
Home › Models › Image Classification › AdamCodd/vit-base-nsfw-detector
V

AdamCodd/vit-base-nsfw-detector

Image Classification·AdamCodd· 641.3K· 80
transformers.js apache-2.0 86.1M params base_model:google/vit-base-patch16-384base_model:quantized:google/vit-base-patch16-384license:apache-2.0deploy:azure

This model is a fine-tuned version of vit-base-patch16-384 on around 25000 images (drawings, photos...). It achieves the following results on the evaluation set: - Loss: 0.0937 - Accuracy: 0.9654

Open in MLForge Sign up free Desktop app Source ↗
# pull & run locally
pip install mlforge-sdk && mlforge pull AdamCodd/vit-base-nsfw-detector

Model details

Task
Image Classification
Provider
AdamCodd
Framework
transformers.js
Parameters
86.1M
Size
1.5 GB
License
apache-2.0
Downloads
641.3K
Likes
80
Updated
2024-12-03

About AdamCodd/vit-base-nsfw-detector

This model is a fine-tuned version of vit-base-patch16-384 on around 25000 images (drawings, photos...). It achieves the following results on the evaluation set: - Loss: 0.0937 - Accuracy: 0.9654

Related Image Classification

M timm/mobilenetv3_small_100.lamb_in1k Image Classification ·2.6M params 22.7M 80 🤗 HF N Falconsai/nsfw_image_detection Image Classification ·85.8M params 9.8M 1.1K 🤗 HF F dima806/fairface_age_image_detection Image Classification ·85.8M params 6.6M 75 🤗 HF V google/vit-base-patch16-224 Image Classification ·86.6M params 5.7M 980 🤗 HF R timm/resnet50.a1_in1k Image Classification 2.6M 43 🤗 HF M apple/mobilevit-small Image Classification 2.4M 91 🤗 HF