Explore Models

By Task
All Models (32)Object Detection Models (19)Classification Models (7)Instance Segmentation Models (3)Semantic Segmentation Models (1)Keypoint Detection Models (2)
By Feature
Foundation Vision ModelsMultimodal Vision ModelsTop LLMs with Vision CapabilitiesImage Embedding ModelsReal Time Vision ModelsZero-Shot Detection ModelsImage Captioning ModelsImage Similarity ModelsImage Tagging ModelsVisual Question Answering ModelsOptimal for Edge Deployment

Top PyTorch Image Classification Models

Explore image classification models that use the PyTorch framework.
‍
Deploy select models (i.e. YOLOv8, CLIP) using the Roboflow Hosted API, or your own hardware using Roboflow Inference.

Classification
Classification
Deploy on Device with Roboflow✅
Classification
Classification
YOLOv8 Classification

Model Size:

MB

Parameters:

68200000

Architecture:

YOLO

An image classification model built using YOLOv8. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide
Classification
Classification
Deploy on Device with Roboflow✅
Classification
Classification
Vision Transformer

Model Size:

MB

Parameters:

Architecture:

The Vision Transformer leverages powerful natural language processing embeddings (BERT) and applies them to images. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide
Classification
Classification
Deploy on Device with Roboflow✅
Classification
Classification
OpenAI CLIP

Model Size:

MB

Parameters:

428000000.0

Architecture:

CLIP (Contrastive Language-Image Pre-Training) is an impressive multimodal zero-shot image classifier that achieves impressive results in a wide range of domains with no fine-tuning. It applies the recent advancements in large-scale transformers like GPT-3 to the vision arena. Learn more »
How to train
Colab notebook
Jupyter notebook
View on GitHub
Video guide

Deploy a computer vision model today

Join 250,000+ developers curating high quality datasets and deploying better models with Roboflow.

Get started