xiaodongyang

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

LLM101n

LLM101n: Let's build a Storyteller

flash-attention

Fast and memory-efficient exact attention

mit-deep-learning

Tutorials, assignments, and competitions for MIT Deep Learning related courses.

hallo

Hallo: Hierarchical Audio-Driven Visual Synthesis for Portrait Image Animation

fiftyone

The open-source tool for building high-quality datasets and computer vision models

pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

nvitop

An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.

BEVFormer

[ECCV 2022] This is the official implementation of BEVFormer, a camera-only framework for autonomous driving perception, e.g., 3D object detection and semantic map segmentation.

matplotlib-cheatsheet

Matplotlib 3.1 cheat sheet.

SensorsCalibration

OpenCalib: A Multi-sensor Calibration Toolbox for Autonomous Driving

bevfusion

[ICRA'23] BEVFusion: Multi-Task Multi-Sensor Fusion with Unified Bird's-Eye View Representation

Transformers-Recipe

🧠 A study guide to learn about Transformers

BEVDet

Official code base of the BEVDet series .

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

simtrack

Exploring Simple 3D Multi-Object Tracking for Autonomous Driving (ICCV 2021)

VehicleX

VehicleX: Simulating Content Consistent Vehicle Datasets with Attribute Descent (ECCV 2020, TPAMI 2023)

CODD

Cooperative Driving Dataset: a dataset for multi-agent driving scenarios

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)

gedepth

GEDepth: Ground Embedding for Monocular Depth Estimation (ICCV 2023)

Awesome-Synthetic-Data-for-Perception-Task

tip

Transcendental Idealism of Planner: Evaluating Perception from Planning Perspective for Autonomous Driving (ICML 2023)