Implementation of "Unsupervised Domain Adaptive 3D Detection with Multi-Level Consistency"

Last update: Nov 17, 2022

Related tags

Overview

Unsupervised Domain Adaptive 3D Detection with Multi-Level Consistency (ICCV2021)

Paper Link: https://arxiv.org/abs/2107.11355

This implementation builds on top of OpenPCDet. Please kindly refer to its Github repo for introduction and recent updates.

Installation

a. Set up the environment:

Python 3.6+ (3.7 suggested)
Pytorch 1.1+ (1.3 suggested)

b. Clone this repository.

c. Install requirements

pip install -r requirements.txt

d. Install the pcdet library:

python setup.py develop

e. Data preparation

Please refer to GETTING_STARTED.md for data preparation and basic usage of pcdet .

Training and Testing

a. Train a base model on the source domain (KITTI) as the pretrained model for domain adaptation

cd tools/

sh scripts/dist_train.sh ${NUM_GPUS} --cfg_file cfgs/kitti_models/pointrcnn.yaml

# or using slurm
sh scripts/slurm_train.sh ${PARTITION} ${JOB_NAME} ${NUM_GPUS} --cfg_file cfgs/kitti_models/pointrcnn.yaml

b. Train the domain adaptation model

sh scripts/dist_train_mean_teacher.sh ${NUM_GPUS} \
--cfg_file cfgs/kitti_models/pointrcnn_mean_teacher_waymo.yaml \
--pretrained_model ${PATH_TO_PRETRAINED_MODEL_CHECKPOINT}

# or using slurm
sh scripts/slurm_train_mean_teacher.sh ${PARTITION} ${JOB_NAME} ${NUM_GPUS} \
--cfg_file cfgs/kitti_models/pointrcnn_mean_teacher_waymo.yaml \
--pretrained_model ${PATH_TO_PRETRAINED_MODEL_CHECKPOINT}

c. Test on the target domain (Waymo)

python test.py --cfg_file cfgs/waymo_models/pointrcnn_kitti2waymo.yaml \
--ckpt ${PATH_TO_MODEL_CHECKPOINT}

Implementation of "Unsupervised Domain Adaptive 3D Detection with Multi-Level Consistency"

Related tags

Overview

Unsupervised Domain Adaptive 3D Detection with Multi-Level Consistency (ICCV2021)

Installation

Training and Testing

Owner

Lighthouse: Predicting Lighting Volumes for Spatially-Coherent Illumination

Inflated i3d network with inception backbone, weights transfered from tensorflow

Tools for manipulating UVs in the Blender viewport.

A Python package for causal inference using Synthetic Controls

An NVDA add-on to split screen reader and audio from other programs to different sound channels

Video Background Music Generation with Controllable Music Transformer (ACM MM 2021 Oral)

Pytorch implementation of the paper "Topic Modeling Revisited: A Document Graph-based Neural Network Perspective"

EfficientMPC - Efficient Model Predictive Control Implementation

The Power of Scale for Parameter-Efficient Prompt Tuning

Project for music generation system based on object tracking and CGAN

Interacting Two-Hand 3D Pose and Shape Reconstruction from Single Color Image (ICCV 2021)

ViewFormer: NeRF-free Neural Rendering from Few Images Using Transformers

This is the code of paper ``Contrastive Coding for Active Learning under Class Distribution Mismatch'' with python.

TensorFlow code for the neural network presented in the paper: "Structural Language Models of Code" (ICML'2020)

Py4fi2nd - Jupyter Notebooks and code for Python for Finance (2nd ed., O'Reilly) by Yves Hilpisch.

Pytorch implementation of Hinton's Dynamic Routing Between Capsules

A minimal solution to hand motion capture from a single color camera at over 100fps. Easy to use, plug to run.

Torch implementation of "Enhanced Deep Residual Networks for Single Image Super-Resolution"

Voxel Set Transformer: A Set-to-Set Approach to 3D Object Detection from Point Clouds (CVPR 2022)

Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch