dingwoai

nerfacc

A General NeRF Acceleration Toolbox in PyTorch.

English-Writing

Enhance Your English Writing for Science Research 写论文英语素材

SSL-Lanes

[CoRL-2022] SSL-Lanes: Self-Supervised Learning for Motion Forecasting in Autonomous Driving

SensorsCalibration

OpenCalib: A Multi-sensor Calibration Toolbox for Autonomous Driving

nuscenes-devkit

The devkit of the nuScenes dataset.

vista

Data-driven simulation for training and evaluating full-scale autonomous vehicles.

HandyFigure

HandyFigure provides the sources file (ususally PPT files) for paper figures

moai

moai is a PyTorch-based AI Model Development Kit (MDK) created to improve data-driven model workflows, design and reproducibility.

camviz

Visualization Library

vidt

scalene

Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals

How_to_optimize_in_GPU

This is a series of GPU optimization topics. Here we will introduce how to optimize the CUDA kernel in detail. I will introduce several basic kernel optimizations, including: elementwise, reduce, sgemv, sgemm, etc. The performance of these kernels is basically at or near the theoretical limit.

multiview_notebooks

A collection of educational notebooks on multi-view geometry and computer vision.

BMTrain

Efficient Training (including pre-training and fine-tuning) for Big Models

python_plot_utils

A simple code for plotting figure, colorbar, and cropping with python

calibration_kit

TransFusion

[PyTorch] Official implementation of CVPR2022 paper "TransFusion: Robust LiDAR-Camera Fusion for 3D Object Detection with Transformers". https://arxiv.org/abs/2203.11496

mono-semantic-maps

xaitk-saliency

As part of the Explainable AI Toolkit (XAITK), XAITK-Saliency is an open source, explainable AI framework for visual saliency algorithm interfaces and implementations, built for analytics and autonomy applications.

pyrelational

pyrelational is a python active learning library for rapidly implementing active learning pipelines from data management, model development (and Bayesian approximation), to creating novel active learning strategies.

pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

ffcv

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

yt-dlp

A feature-rich command-line audio/video downloader

transformer-quantization

kitti360LabelTool

LOST

Pytorch implementation of LOST unsupervised object discovery method

MuarAugment

State-of-the-art data augmentation search algorithms in PyTorch

awesome-lane-detection

A paper list of lane detection.

MonoFlex

Released code for Objects are Different: Flexible Monocular 3D Object Detection, CVPR21

xmodaler

X-modaler is a versatile and high-performance codebase for cross-modal analytics(e.g., image captioning, video captioning, vision-language pre-training, visual question answering, visual commonsense reasoning, and cross-modal retrieval).