Repository Index

Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

• ultralytics
• yolov8
• yolo
• deep-learning

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

• yolov5
• object-detection
• pytorch
• onnx

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

• pytorch
• resnet
• pretrained-models
• pretrained-weights

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

• artificial-intelligence
• attention-mechanism
• transformers
• computer-vision

CVAT is a leading data annotation platform for image, video, audio, and 3D datasets. It offers open-source, cloud, and enterprise products, as well as labeling services, with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

• video-annotation
• computer-vision
• computer-vision-annotation
• image-annotation

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

• deep-learning
• pytorch
• grad-cam
• visualizations

X-AnyLabeling: A lightweight, efficient, and unified cross-platform desktop application for annotating text, image, video, point cloud and multimodal data, combining versatile built-in tools with state-of-the-art AI models and flexible multi-format export.

• sam
• yolo
• deep-learning
• object-detection

Translated[CVPR 2024 Oral] InternVL Family: A Pioneering Open-Source Alternative to GPT-4o. An open-source multimodal dialogue model with performance close to GPT-4o

• image-classification
• image-text-retrieval
• llm
• semantic-segmentation

Gluon CV Toolkit

• deep-learning
• computer-vision
• neural-network
• gluon

A treasure chest for visual classification and recognition powered by PaddlePaddle

• image-classification
• knowledge-distillation
• autoaugment
• randaugment

An absolute beginner's guide to Machine Learning and Image Classification with Neural Networks

• neural-network
• image-classification
• machine-learning
• caffe

Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet, NasNet, Residual Attention Network, SENet, WideResNet)

• pytorch
• image-classification
• deep-learning
• cifar100

Differentiable architecture search for convolutional and recurrent networks

• deep-learning
• automl
• image-classification
• language-modeling

OpenMMLab Pre-training Toolbox and Benchmark

• image-classification
• resnet
• mobilenet
• pytorch

A library for transfer learning by reusing parts of TensorFlow models.

• tensorflow
• machine-learning
• transfer-learning
• embeddings

Accelerated deep learning R&D

• deep-learning
• reinforcement-learning
• machine-learning
• computer-vision

Sandbox for training deep learning networks

• machine-learning
• deep-learning
• mxnet
• gluon

Images to inference with no labeling (use foundation models to train supervised models).

• computer-vision
• model-distillation
• auto-labeling
• deep-learning

The collection of pre-trained, state-of-the-art AI models for ailia SDK

• deep-learning
• face-recognition
• face-detection
• object-detection

A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.

• pytorch
• imbalanced-data
• image-classification
• data-sampling

[CVPR 2025] Official PyTorch Implementation of MambaVision: A Hybrid Mamba-Transformer Vision Backbone

• deep-learning
• foundation-models
• image-classification
• mamba

Implementation of EfficientNet model. Keras and TensorFlow Keras.

• classification
• imagenet
• efficientnet
• pretrained-models

Use of Attention Gates in a Convolutional Neural Network / Medical Image Classification and Segmentation

• convolutional-neural-networks
• attention-model
• image-classification
• image-segmentation

Integrate deep learning models for image classification | Backbone learning/comparison/magic modification project

• pytorch
• image-classification
• transformer
• cnn