Top PyTorch Object Detection Models

Explore object detection 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.

Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

YOLO

YOLOv9 is an object detection model architecture released on February 21st, 2024. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

DocTR is an Optical Character Recognition tool powered by deep learning. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

YOLO

YOLO-NAS is an object detection model developed by Deci that achieves SOTA performances compared to YOLOv5, v7, and v8. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

Grounding DINO is a zero-shot object detection model made by combining a Transformer-based DINO detector and grounded pre-training. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

Transformers

Detection Transformer (DETR) is an end-to-end object detection model implemented using the Transformer architecture. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

YOLO, CNN

YOLOv8 is a state-of-the-art object detection and image segmentation model created by Ultralytics, the developers of YOLOv5. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

68.7

MB

Parameters:

9 million

Top FPS:

Architecture:

CNN, YOLO

YOLOX is a high-performance object detection model. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

202.0

MB

Parameters:

12,786,711 (S2D)

Top FPS:

106

Architecture:

CNN, YOLO

YOLOR (You Only Learn One Representation) is an object detection model that uses both implicit and explicit knowledge to make predictions. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

Transformer, YOLO

YOLOS looks at patches of an image to to form "patch tokens", which are used in place of the traditional wordpiece tokens in NLP. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

YOLO

Scaled YOLOv4 is an extension of the YOLOv4 research implemented in the YOLOv5 PyTorch framework. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

75.6

MB

Parameters:

Top FPS:

161

Architecture:

YOLO, CNN

YOLOv7 is a state of the art object detection model. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

520

Architecture:

CNN, YOLO

MT-YOLOv6 is a YOLO based model released in 2022. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

7.5 million

Top FPS:

Architecture:

CNN, YOLO

YOLOv5-OBB is a variant of YOLOv5 that supports oriented bounding boxes. This model is designed to yield predictions that better fit objects that are positioned at an angle. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

YOLO

YOLOv4 has emerged as the best real time object detection model. YOLOv4 carries forward many of the research contributions of the YOLO family of models along with new modeling and data augmentation techniques. This implementation is in PyTorch. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

YOLO

Though it is no longer the most accurate object detection algorithm, YOLO v3 is still a very good choice when you need real-time detection while maintaining excellent accuracy. PyTorch version. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

3.9 million

Top FPS:

97

Architecture:

EfficientDet achieves the best performance in the fewest training epochs among object detection model architectures, making it a highly scalable architecture especially when operating with limited compute. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

MB

Parameters:

Top FPS:

Architecture:

Detectron2 is model zoo of it's own for computer vision models written in PyTorch. Learn more »
Object Detection
Object Detection
Deploy on Device with Roboflow✅
Object Detection
Object Detection

Model Size:

14.1

MB

Parameters:

7.2 million

Top FPS:

140

Architecture:

CNN, YOLO

A very fast and easy to use PyTorch model that achieves state of the art (or near state of the art) results. Learn more »

Deploy a computer vision model today

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

Get started