OpenMMLab Computer Vision Foundation

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Implementing a ChatGPT-like LLM from scratch, step by step

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

🧑‍🏫 59 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, ...), gans(cycleg…

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeX…

YOLOv6: a single-stage object detection framework dedicated to industrial applications.

OpenMMLab Pre-training Toolbox and Benchmark

A PyTorch implementation of the Transformer model in "Attention is All You Need".

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

End-to-End Object Detection with Transformers

VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

An all-in-one toolkit for computer vision

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image …

NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥

Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.

An anomaly detection library comprising state-of-the-art algorithms and features such as experiment management, hyper-parameter optimization, and edge inference.

CnOCR: Awesome Chinese/English OCR toolkits based on PyTorch/MXNet, It comes with 20+ well-trained models for different application scenarios and can be used directly after installation. 【基于 PyTorc…

Contains the "pycocotools" package on PyPI. Changes made to the official cocoapi about packaging.

OpenMMLab Foundational Library for Training Deep Learning Models

Efficient computing methods developed by Huawei Noah's Ark Lab

OpenMMLab YOLO series toolbox and benchmark