AdamCodd / image-classification updated 1 year ago

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

Params

90 M

Context

—

Downloads 30d

660 K

Likes

81

Commercial use: allowed apache-2.0 Not gated SAFETENSORS View on Hugging Face ↗

Download history

daily snapshots · 10 days

661 K638 K

Jul 28Jul 31Aug 3Aug 6

Can you run it?

Estimated VRAM at 8K context unless noted. Pick your hardware to see the verdict per quantization.

File	Quant	Size	Est. VRAM	Verdict on RTX 4090 · 24 GB
model.safetensors	f32	0.3 GB	0.9 GB	✅ Runs comfortably

Estimate: file size × 1.1 + KV cache at 8K + 0.5 GB overhead. Not a benchmark — how we calculate this.

Specifications

Architecture: ViTForImageClassification
Parameters: 90 M
Tensor type: F32
Layers / heads: 12 / 12
Licence: apache-2.0
First seen on the Hub: 2024-01-03
Training datasets: undisclosed
AUC (reported): 0.9948
Loss (reported): 0.0937
Accuracy (reported): 0.9654
Added to our catalog: 2026-07-28

Compare with

mobilenetv3_small_100.lamb_in1k 18.5 M nsfw_image_detection 5.4 M vit-base-patch16-224 4.7 M fairface_age_image_detection 3.2 M mit-b2 1.6 M gender-classification 1.5 M

Appears in

Best open coding models Commercial-use text models Models that fit in 12 GB