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



Colab notebook

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 »

Multimodal Model

fEATURED

Multimodal Model

CogVLM

Model Size:

Parameters:

Top FPS:

Architecture:

that shows strong performanve in Visual Question Answering (VQA) and other vision tasks. CogVLM achieves state-of-the-art performance on a range of vision benchmarks. 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 »

Instance Segmentation

fEATURED

Instance Segmentation

YOLACT

Model Size:

Parameters:

Top FPS:

Architecture:

A simple, fully convolutional model for real-time instance segmentation 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

Instance Segmentation

fEATURED

Instance Segmentation

FastSAM

Model Size:

Parameters:

Top FPS:

Architecture:

FastSAM is an image segmentation model trained using 2% of the data in the Segment Anything Model SA-1B dataset. Learn more »

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 »

Multimodal Model

fEATURED

Multimodal Model

QwenVL

Model Size:

Parameters:

Top FPS:

Architecture:

Qwen-VL is an LMM developed by Alibaba Cloud. Qwen-VL accepts images, text, and bounding boxes as inputs. The model can output text and bounding boxes. Qwen-VL naturally supports English, Chinese, and multilingual conversation. 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

Classification

ResNet 34

Model Size:

Parameters:

Top FPS:

Architecture:

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

Object Detection

fEATURED

Object Detection

YOLOv4 Tiny

Model Size:

Parameters:

Top FPS:

Architecture:

ResNet-D, YOLO

The tiny and fast version of YOLOv4 - good for training and deployment on limited compute resources, and getting a feel for your dataset Learn more »

Object Detection

fEATURED

Object Detection

ByteTrack

Model Size:

Parameters:

Top FPS:

Architecture:

ByteTrack is a multi-object tracking computer vision model. Learn more »

Object Detection

fEATURED

Object Detection

RTMDet

Model Size:

Parameters:

Top FPS:

Architecture:

RTMDet is an efficient real-time object detector, with self-reported metrics outperforming the YOLO series. It achieves 52.8% AP on COCO with 300+ FPS on an NVIDIA 3090 GPU, making it one of the fastest and most accurate object detectors available as of writing this post. Learn more »

Object Detection

fEATURED

Object Detection

DINOv2

Model Size:

Parameters:

Top FPS:

Architecture:

DINOv2 is a self-supervised method for training computer vision models developed by Meta Research and released in April 2023. Learn more »

Object Detection

fEATURED

Object Detection

DocTR

Model Size:

Parameters:

Top FPS:

Architecture:

DocTR is an Optical Character Recognition tool powered by deep learning. Learn more »

Object Detection

fEATURED

Object Detection

L2CS-Net

Model Size:

Parameters:

Top FPS:

Architecture:

L2CS-Net is a gaze estimation model that enables you to calculate where someone is looking and in what direction someone is looking. Learn more »

Object Detection

fEATURED

Object Detection

Kosmos-2

Model Size:

Parameters:

Top FPS:

Architecture:

Kosmos-2 is a multimodal language model capable of object detection and grounding text in images. Learn more »

Object Detection

fEATURED

Object Detection

LLaVA

Model Size:

Parameters:

Top FPS:

Architecture:

LLaVA is an open source multimodal language model that you can use for visual question answering and has limited support for object detection. Learn more »

Object Detection

fEATURED

Object Detection

OWLv2

Model Size:

Parameters:

Top FPS:

Architecture:

OWLv2 is a transformer-based object detection model developed by Google Research. OWLv2 is the successor to OWL ViT. Learn more »

Classification

fEATURED

Classification

MetaCLIP

Model Size:

Parameters:

Top FPS:

Architecture:

MetaCLIP is a zero-shot classification and embedding model developed by Meta AI. Learn more »

Classification

fEATURED

Classification

FastViT

Model Size:

Parameters:

Top FPS:

Architecture:

Object Detection

fEATURED

Object Detection

OWL ViT

Model Size:

Parameters:

Top FPS:

Architecture:

OWL-ViT is a transformer-based object detection model developed by Google Research. Learn more »

Classification

fEATURED

Classification

ALBEF

Model Size:

Parameters:

Top FPS:

Architecture:

Classification

fEATURED

Classification

BLIPv2

Model Size:

Parameters:

Top FPS:

Architecture:

Instance Segmentation

fEATURED

Instance Segmentation

SAM-CLIP

Model Size:

Parameters:

Top FPS:

Architecture:

Instance Segmentation

fEATURED

Instance Segmentation

Grounded SAM

Model Size:

Parameters:

Top FPS:

Architecture:

Classification

fEATURED

Classification

BLIP

Model Size:

Parameters:

Top FPS:

Architecture:

Object Detection

fEATURED

Object Detection

Grounding DINO

Model Size:

Parameters:

Top FPS:

Architecture:

Grounding DINO is a state-of-the-art zero-shot object detection model, developed by IDEA Research. Learn more »

Object Detection

fEATURED

Object Detection

GPT-4V

Model Size:

Parameters:

Top FPS:

Architecture:

Object Detection

fEATURED

Object Detection

CoDet

Model Size:

Parameters:

Top FPS:

Architecture:

CoDet is an open vocabulary zero-shot object detection model. Learn more »

Object Detection

fEATURED

Object Detection

VLPart

Model Size:

Parameters:

Top FPS:

Architecture:

VLPart, developed by Meta Research, is an object detection and segmentation model that works with an open vocabulary Learn more »

Multimodal Model

fEATURED

Multimodal Model

BakLLaVA

Model Size:

Parameters:

Top FPS:

Architecture:

BakLLaVA is an LMM developed by LAION, Ontocord, and Skunkworks AI. BakLLaVA uses a Mistral 7B base augmented with the LLaVA 1.5 architecture. Learn more »