Stars
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), ga…
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Python sample codes for robotics algorithms.
State-of-the-art 2D and 3D Face Analysis Project
Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).
Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.
Geometric Computer Vision Library for Spatial AI
PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
Text-to-3D & Image-to-3D & Mesh Exportation with NeRF + Diffusion.
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
The TinyLlama project is an open endeavor to pretrain a 1.1B Llama model on 3 trillion tokens.
[CVPR 2024] Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data. Foundation Model for Monocular Depth Estimation
BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models
[ECCV 2024] Official implementation of the paper "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"
[CVPR 2024] Real-Time Open-Vocabulary Object Detection
[NeurIPS 2023] Official implementation of the paper "Segment Everything Everywhere All at Once"
⛹️ Pytorch ReID: A tiny, friendly, strong pytorch implement of person re-id / vehicle re-id baseline. Tutorial 👉https://github.com/layumi/Person_reID_baseline_pytorch/tree/master/tutorial
Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
Framework agnostic sliced/tiled inference + interactive ui + error analysis plots
Denoising Diffusion Probabilistic Models
Simplest working implementation of Stylegan2, state of the art generative adversarial network, in Pytorch. Enabling everyone to experience disentanglement
PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.
min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch
[CVPR2019] Fast Online Object Tracking and Segmentation: A Unifying Approach
StudioGAN is a Pytorch library providing implementations of representative Generative Adversarial Networks (GANs) for conditional/unconditional image generation.
LightGlue: Local Feature Matching at Light Speed (ICCV 2023)