AI Tools.

Search

image classification models

35 models · ranked by HuggingFace downloads

mobilenetv3_small_100.lamb_in1k

MobileNetV3 small model at 100% width multiplier, trained on ImageNet-1k using the LAMB optimizer via the timm library. At under 3M parameters, it targets image classification on mobile and edge hardware where latency and memory are primary constraints. Part of timm's standardized pretrained model zoo with consistent preprocessing and inference APIs.

22,549,780 ↓ · 66 ♡

nsfw_image_detection

Vision Transformer (ViT) fine-tuned for binary NSFW vs. safe image classification. Provides a single classifier for flagging potentially unsafe image content without category-level labeling. Built on ViT-base architecture and fine-tuned on a curated dataset of safe and unsafe images.

21,530,509 ↓ · 1,065 ♡

fairface_age_image_detection

A ViT-base model fine-tuned on the FairFace dataset for age bracket classification from face images. It categorizes detected faces into age groups (0-2, 3-9, 10-19, 20-29, 30-39, 40-49, 50-59, 60-69, 70+). Built on google/vit-base-patch16-224-in21k and fine-tuned with Apache 2.0 license.

6,277,118 ↓ · 73 ♡

vit-base-patch16-224

Google's ViT-Base (Vision Transformer base model) with 16×16 pixel patch size trained at 224px resolution on ImageNet-21k and fine-tuned on ImageNet-1k. The paper introducing ViTs demonstrated that pure transformer architectures without convolutional inductive bias can match CNNs on image classification when trained on sufficient data. Widely used as a starting backbone for image classification fine-tuning.

4,785,312 ↓ · 957 ♡

nsfw-image-detection-384

A fine-tuned image classifier from Marqo that flags adult or explicit content in images at 384px input resolution. It outputs probability scores for NSFW versus safe content and is commonly used as a pre-filter in content moderation pipelines before storing or serving user uploads. Apache 2.0 licensed for commercial deployment.

3,565,970 ↓ · 52 ♡

mobilevit-small

mobilevit-small is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

2,893,367 ↓ · 91 ♡

convnextv2_nano.fcmae_ft_in22k_in1k

convnextv2_nano.fcmae_ft_in22k_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

1,705,289 ↓ · 4 ♡

resnet18.a1_in1k

resnet18.a1_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

1,551,276 ↓ · 14 ♡

resnet50.a1_in1k

resnet50.a1_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

1,549,910 ↓ · 41 ♡

vit-base-nsfw-detector

vit-base-nsfw-detector is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

1,479,148 ↓ · 75 ♡

gender-classification

gender-classification is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

1,223,055 ↓ · 60 ♡

efficientnet_b0.ra_in1k

efficientnet_b0.ra_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

1,093,337 ↓ · 9 ♡

CommunityForensics-DeepfakeDet-ViT

CommunityForensics-DeepfakeDet-ViT is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

794,856 ↓ · 12 ♡

nsfw_image_detector

nsfw_image_detector is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

683,320 ↓ · 57 ♡

rorshark-vit-base

rorshark-vit-base is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

654,453 ↓ · 3 ♡

resnet18.fb_swsl_ig1b_ft_in1k

resnet18.fb_swsl_ig1b_ft_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

615,141 ↓ · 0 ♡

resnet34.a1_in1k

resnet34.a1_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

609,986 ↓ · 1 ♡

gender_class

gender_class is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

585,400 ↓ · 1 ♡

resnet-18

resnet-18 is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

538,900 ↓ · 67 ♡

wide_resnet50_2.racm_in1k

wide_resnet50_2.racm_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

522,984 ↓ · 2 ♡

vit_base_patch16_224.augreg2_in21k_ft_in1k

vit_base_patch16_224.augreg2_in21k_ft_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

498,644 ↓ · 13 ♡

tf_efficientnetv2_s.in21k_ft_in1k

tf_efficientnetv2_s.in21k_ft_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

467,987 ↓ · 2 ♡

edgenext_small.usi_in1k

edgenext_small.usi_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

439,600 ↓ · 6 ♡

vit_tiny_r_s16_p8_224.augreg_in21k

vit_tiny_r_s16_p8_224.augreg_in21k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

436,469 ↓ · 0 ♡

swinv2-tiny-patch4-window16-256

swinv2-tiny-patch4-window16-256 is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

424,746 ↓ · 13 ♡

vit-base-violence-detection

vit-base-violence-detection is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

414,673 ↓ · 10 ♡

open-age-detection

open-age-detection is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

414,321 ↓ · 6 ♡

deit-base-distilled-patch16-384

deit-base-distilled-patch16-384 is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

412,886 ↓ · 8 ♡

convnext_femto.d1_in1k

convnext_femto.d1_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

406,856 ↓ · 1 ♡

convnext_tiny.in12k_ft_in1k

convnext_tiny.in12k_ft_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

394,431 ↓ · 5 ♡

vit_tiny_patch16_224.augreg_in21k_ft_in1k

vit_tiny_patch16_224.augreg_in21k_ft_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

357,548 ↓ · 3 ♡

beit-base-patch16-224

beit-base-patch16-224 is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

315,859 ↓ · 9 ♡

vit_base_patch16_224.augreg_in21k

vit_base_patch16_224.augreg_in21k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

315,334 ↓ · 11 ♡

resnet-50

resnet-50 is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

314,627 ↓ · 492 ♡

test_resnet.r160_in1k

test_resnet.r160_in1k is an open-source image-classification model available on HuggingFace. Details are sourced from the public model registry.

287,912 ↓ · 0 ♡