The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

End-to-End Object Detection with Transformers

Fast Segment Anything

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

FFCV: Fast Forward Computer Vision (and other ML workloads!)

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Object Detection and Instance Segmentation.

Tutorial for building a custom CUDA function for Pytorch

Model Zoo of BDD100K Dataset

[ICLR 2025] Diffusion Feedback Helps CLIP See Better

Official implementation of "CrossPoint: Self-Supervised Cross-Modal Contrastive Learning for 3D Point Cloud Understanding" (CVPR, 2022)

Code for "Deep Marching Cubes: Learning Explicit Surface Representations", CVPR 2018

Official code of the paper "Expanding Low-Density Latent Regions for Open-Set Object Detection" (CVPR 2022)

Official code for RbA: Segmenting Unknown Regions Rejected by All (ICCV 2023)

SLAMP: Stochastic Latent Appearance and Motion Prediction

Code for "Multi-Object Discovery by Low-Dimensional Object Motion"