Xiaodong Yang's starred repositories
pytorch-image-models
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
flash-attention
Fast and memory-efficient exact attention
mit-deep-learning
Tutorials, assignments, and competitions for MIT Deep Learning related courses.
pytorch-OpCounter
Count the MACs / FLOPs of your PyTorch model.
matplotlib-cheatsheet
Matplotlib 3.1 cheat sheet.
SensorsCalibration
OpenCalib: A Multi-sensor Calibration Toolbox for Autonomous Driving
Transformers-Recipe
🧠A study guide to learn about Transformers
data-centric-AI
A curated, but incomplete, list of data-centric AI resources.
SegmenTron
Support PointRend, Fast_SCNN, HRNet, Deeplabv3_plus(xception, resnet, mobilenet), ContextNet, FPENet, DABNet, EdaNet, ENet, Espnetv2, RefineNet, UNet, DANet, HRNet, DFANet, HardNet, LedNet, OCNet, EncNet, DuNet, CGNet, CCNet, BiSeNet, PSPNet, ICNet, FCN, deeplab)
nuplan-devkit
The devkit of the nuPlan dataset.
pillarnext
PillarNeXt: Rethinking Network Designs for 3D Object Detection in LiDAR Point Clouds (CVPR 2023)
LiDAR_snow_sim
LiDAR snowfall simulation
Learning-to-See-Moving-Objects-in-the-Dark
Learning to See Moving Objects in the Dark. ICCV 2019
pillar-motion
Self-Supervised Pillar Motion Learning for Autonomous Driving (CVPR 2021)
distill-bev
DistillBEV: Boosting Multi-Camera 3D Object Detection with Cross-Modal Knowledge Distillation (ICCV 2023)
info-ground
Learning phrase grounding from captioned images through InfoNCE bound on mutual information
MANTRA-CVPR20
Official Pytorch code for MANTRA - Memory Augmented Neural Trajectory Predictor (CVPR2020)