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

The code for 'Deep Residual Fourier Transformation for Single Image Deblurring'

A motion tracking system for any arbitaray points in a video frame.

Differentiable Annealed Importance Sampling (DAIS)

Pynomial - a lightweight python library for implementing the many confidence intervals for the risk parameter of a binomial model

Official Pytorch Implementation for Splicing ViT Features for Semantic Appearance Transfer presenting Splice

GPU Programming with Julia - course at the Swiss National Supercomputing Centre (CSCS), ETH Zurich

EGNN - Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch

A collection of Jupyter notebooks to play with NVIDIA's StyleGAN3 and OpenAI's CLIP for a text-based guided image generation.

Joint Learning of 3D Shape Retrieval and Deformation, CVPR 2021

Build tensorflow keras model pipelines in a single line of code. Created by Ram Seshadri. Collaborators welcome. Permission granted upon request.

BARF: Bundle-Adjusting Neural Radiance Fields 🤮 (ICCV 2021 oral)

Fast SHAP value computation for interpreting tree-based models

Code for "Unsupervised Layered Image Decomposition into Object Prototypes" paper

Robot Servers and Server Manager software for robo-gym

Codes to pre-train T5 (Text-to-Text Transfer Transformer) models pre-trained on Japanese web texts

A command line simple note taking app

D2LV: A Data-Driven and Local-Verification Approach for Image Copy Detection

ManiSkill-Learn is a framework for training agents on SAPIEN Open-Source Manipulation Skill Challenge (ManiSkill Challenge), a large-scale learning-from-demonstrations benchmark for object manipulation.

PyBullet CartPole and Quadrotor environments—with CasADi symbolic a priori dynamics—for learning-based control and reinforcement learning

FirmWire is a full-system baseband firmware emulation platform for fuzzing, debugging, and root-cause analysis of smartphone baseband firmwares