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

PyTorch reimplementation of the paper Involution: Inverting the Inherence of Convolution for Visual Recognition [CVPR 2021].

Research code of ICCV 2021 paper "Mesh Graphormer"

QuadTree Attention for Vision Transformers (ICLR2022)

Visual Adversarial Imitation Learning using Variational Models (VMAIL)

Code and real data for the paper "Counterfactual Temporal Point Processes", available at arXiv.

Official implementation for "Image Quality Assessment using Contrastive Learning"

Open source repository for the code accompanying the paper 'Non-Rigid Neural Radiance Fields Reconstruction and Novel View Synthesis of a Deforming Scene from Monocular Video'.

Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia and Matlab

Export CenterPoint PonintPillars ONNX Model For TensorRT

A copy of Ares that costs 30 fucking dollars.

Code reproduce for paper "Vehicle Re-identification with Viewpoint-aware Metric Learning"

Vision transformers (ViTs) have found only limited practical use in processing images

[AAAI 2022] Separate Contrastive Learning for Organs-at-Risk and Gross-Tumor-Volume Segmentation with Limited Annotation

ParaGen is a PyTorch deep learning framework for parallel sequence generation

RuDOLPH: One Hyper-Modal Transformer can be creative as DALL-E and smart as CLIP

Entity-Based Knowledge Conflicts in Question Answering.

Generate saved_model, tfjs, tf-trt, EdgeTPU, CoreML, quantized tflite and .pb from .tflite.

python library for invisible image watermark (blind image watermark)

StyleGAN - Official TensorFlow Implementation

PyTorch experiments with the Zalando fashion-mnist dataset