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

Model Size:

MB

Parameters:

Top FPS:

Architecture:

YOLO

An image classification model built using YOLOv8. Learn more »
Classification
Classification
Deploy on Device with Roboflow✅
Classification
Classification

Model Size:

MB

Parameters:

Top FPS:

Architecture:

The Vision Transformer leverages powerful natural language processing embeddings (BERT) and applies them to images. Learn more »
Classification
Classification
Deploy on Device with Roboflow✅
Classification
Classification

Model Size:

MB

Parameters:

Top FPS:

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 »

Deploy a computer vision model today

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

Get started