Image Classification

Classify images with two lines of code. These models are ready to go, often with pre-trained weights and exports available for mobile or server-side inference.

Looking for a dataset? Explore image classification datasets.

Deploy select models (i.e. YOLOv8, CLIP) using the Roboflow Hosted API, or your own hardware using Roboflow Inference.
Meta

MetaCLIP

MetaCLIP is a zero-shot classification and embedding model developed by Meta AI.
Classification
Deploy with Roboflow
Ultralytics

YOLOv8 Classification

An image classification model built using YOLOv8.
Classification
Deploy with Roboflow
Ultralytics

YOLOv5 Classification

YOLOv5 Classification is a version of the YOLOv5 model used in single-label and multi-label image classification.
Classification
Deploy with Roboflow
Salesforce

BLIP

Classification
Deploy with Roboflow

EfficientNet

EfficientNet is from a family of image classification models from GoogleAI that train comparatively quickly on small amounts of data, making the most of limited datasets.
Classification
Deploy with Roboflow

ResNet 34

A fast, simple convolutional neural network that gets the job done for many tasks, including classification.
Classification
Deploy with Roboflow

ResNet-50

ResNet-50 is a popular image classification model architecture.
Classification
Deploy with Roboflow

AltCLIP

AltCLIP is a zero-shot image classification model.
Classification
Deploy with Roboflow

RemoteCLIP

RemoteCLIP is a zero-shot classification model for remote sensing.
Classification
Deploy with Roboflow

BioCLIP

BioCLIP is a Vision Foundation Model for the Tree of Life
Classification
Deploy with Roboflow
Apple

MobileCLIP

MobileCLIP is an image embedding model developed by Apple and introduced in the "MobileCLIP: Fast Image-Text Models through Multi-Modal Reinforced Training" paper
Classification
Deploy with Roboflow
Google

SigLIP

SigLIP is an image embedding model defined in the "Sigmoid Loss for Language Image Pre-Training" paper.
Classification
Deploy with Roboflow
Salesforce

BLIPv2

BLIPv2 is a multimodal model developed by Salesforce Research.
Classification
Deploy with Roboflow
Salesforce

ALBEF

Classification
Deploy with Roboflow
Apple

FastViT

FastViT is a fast image classification model developed by Apple.
Classification
Deploy with Roboflow

ResNet 32

A fast, simple convolutional neural network that gets the job done for many tasks, including classification.
Classification
Deploy with Roboflow
Google

Vision Transformer

The Vision Transformer leverages powerful natural language processing embeddings (BERT) and applies them to images.
Classification
Deploy with Roboflow

MobileNet V2 Classification

MobileNet is a GoogleAI model well-suited for on-device, real-time classification (distinct from MobileNetSSD, Single Shot Detector). This implementation leverages transfer learning from ImageNet to your dataset.
Classification
Deploy with Roboflow

Frequently Asked Questions

What models are used for image classification?

There are a wide variety of models used for image classification. Popular choices of models for image classification tasks include YOLOv5, the Vision Transformer, and Resnet34.

What are the use cases for image classification?

Image classification is useful in any computer vision task where you need to assign content into one of a limited number of categories. Here are a few examples of real-world use cases for image classification:

  • Deciding whether an image contains explicit material
  • Classifying plant species
  • Identifying wildlife species
  • Tracking what types of vehicles enter a parking lot (i.e. cars, motorbikes)

What is image classification?

Image classification is a computer vision task where images are assigned a label based on their contents. Only one label is assigned per image. For example, consider a dataset that classifies tree species. One photo may be given the class “birch” and another “fir”.

Where Can I Learn More About Object Detection?

View All Learning Resources
No items found.