Top Computer Vision Models

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 »



How to train



Instance Segmentation

fEATURED

Instance Segmentation

YOLOv5 Instance Segmentation

Model Size:

Parameters:

Top FPS:

Architecture:

CNN, YOLO

YOLOv5 Instance Segmentation is a version of YOLOv5 that can be used for instance segmentation tasks. Learn more »

Classification

fEATURED

Classification

YOLOv5 Classification

Model Size:

Parameters:

Top FPS:

Architecture:

YOLOv5 Classification is a version of the YOLOv5 model used in single-label and multi-label image classification. Learn more »

Object Detection

fEATURED

Object Detection

Detectron2

Model Size:

Parameters:

Top FPS:

Architecture:

Detectron2 is model zoo of it's own for computer vision models written in PyTorch. Learn more »

Instance Segmentation

fEATURED

Instance Segmentation

Mask RCNN

Model Size:

Parameters:

Top FPS:

Architecture:

Mask RCNN is a convolutional neural network for instance segmentation. Learn more »

Object Detection

fEATURED

Object Detection

YOLOv4 Darknet

Model Size:

Parameters:

Top FPS:

Architecture:

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 Darknet. Learn more »

Classification

fEATURED

Classification

OpenAI CLIP

Model Size:

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 »

Object Detection

fEATURED

Object Detection

DETR

Model Size:

Parameters:

Top FPS:

Architecture:

Transformers

Detection Transformer (DETR) is an end-to-end object detection model implemented using the Transformer architecture. Learn more »

Instance Segmentation

fEATURED

Instance Segmentation

YOLOv7 Instance Segmentation

Model Size:

Parameters:

Top FPS:

Architecture:

YOLOv7 Instance Segmentation lets you perform segmentation tasks with the YOLOv7 model. Learn more »

Object Detection

fEATURED

Object Detection

Faster R-CNN

Model Size:

Parameters:

Top FPS:

Architecture:

One of the most accurate object detection algorithms but requires a lot of power at inference time. A good choice if you can do processing asynchronously on a server. Learn more »

Object Detection

fEATURED

Object Detection

YOLOX

Model Size:

68.7

Parameters:

9 million

Top FPS:

Architecture:

CNN, YOLO

YOLOX is a high-performance object detection model. Learn more »

Object Detection

fEATURED

Object Detection

YOLOv7

Model Size:

75.6

Parameters:

Top FPS:

161

Architecture:

YOLO, CNN

YOLOv7 is a state of the art object detection model. Learn more »

Classification

fEATURED

Classification

EfficientNet

Model Size:

Parameters:

Top FPS:

Architecture:

CNN

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. Learn more »

Object Detection

fEATURED

Object Detection

YOLOv3 Keras

Model Size:

Parameters:

Top FPS:

Architecture:

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. Keras implementation. Learn more »

Object Detection

fEATURED

Object Detection

YOLOv3 PyTorch

Model Size:

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 »

Classification

fEATURED

Classification

Vision Transformer

Model Size:

Parameters:

Top FPS:

Architecture:

The Vision Transformer leverages powerful natural language processing embeddings (BERT) and applies them to images. Learn more »

Object Detection

fEATURED

Object Detection

MT-YOLOv6

Model Size:

Parameters:

Top FPS:

520

Architecture:

CNN, YOLO

MT-YOLOv6 is a YOLO based model released in 2022. Learn more »

Object Detection

fEATURED

Object Detection

YOLOv4 PyTorch

Model Size:

Parameters:

Top FPS:

Architecture:

YOLO

Object Detection

fEATURED

Object Detection

YOLO-NAS

Model Size:

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

fEATURED

Object Detection

Scaled YOLOv4

Model Size:

Parameters:

Top FPS:

Architecture:

Scaled YOLOv4 is an extension of the YOLOv4 research implemented in the YOLOv5 PyTorch framework. Learn more »

Object Detection

fEATURED

Object Detection

MobileNet SSD v2

Model Size:

Parameters:

Top FPS:

Architecture:

This architecture provides good realtime results on limited compute. It's designed to run in realtime (30 frames per second) even on mobile devices. Learn more »

Object Detection

fEATURED

Object Detection

YOLOR

Model Size:

202.0

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 »

Instance Segmentation

fEATURED

Instance Segmentation

DETIC

Model Size:

Parameters:

Top FPS:

Architecture:

Detic is an open source segmentation model developed by Meta Research and released in 2022. Learn more »

Object Detection

fEATURED

Object Detection

EfficientDet (D7) Tensorflow 2

Model Size:

Parameters:

77 million

Top FPS:

Architecture:

A scalable, state of the art object detection model, implemented here within the TensorFlow 2 Object Detection API. Learn more »

Object Detection

fEATURED

Object Detection

EfficientDet

Model Size:

Parameters:

3.9 million

Top FPS:

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 »

Semantic Segmentation

fEATURED

Semantic Segmentation

SegFormer

Model Size:

Parameters:

Top FPS:

Architecture:

Transformers

SegFormer is a computer vision framework used in semantic segmentation tasks, implemented with transformers. Learn more »

Classification

fEATURED

Classification

YOLOv8 Classification

Model Size:

Parameters:

Top FPS:

Architecture:

An image classification model built using YOLOv8. Learn more »

Object Detection

fEATURED

Object Detection

YOLOS

Model Size:

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 »

Instance Segmentation

fEATURED

Instance Segmentation

OneFormer

Model Size:

Parameters:

219 million

Top FPS:

Architecture:

Transformers

OneFormer is a state-of-the-art multi-task image segmentation framework that is implemented using transformers. Learn more »

Classification

fEATURED

Classification

ResNet 32

Model Size:

Parameters:

460,000

Top FPS:

Architecture:

A fast, simple convolutional neural network that gets the job done for many tasks, including classification. Learn more »

Classification

fEATURED

Classification

MobileNet V2 Classification

Model Size:

Parameters:

Top FPS:

Architecture:

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. Learn more »

Classification

fEATURED