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

Unofficial implementation of "TTNet: Real-time temporal and spatial video analysis of table tennis" (CVPR 2020)

2021 National Underwater Robotics Vision Optics

PyTorch implementation of the Flow Gaussian Mixture Model (FlowGMM) model from our paper

MISSFormer: An Effective Medical Image Segmentation Transformer

💡 Type hints for Numpy

NICE-GAN — Official PyTorch Implementation Reusing Discriminators for Encoding: Towards Unsupervised Image-to-Image Translation

Computer Vision and Pattern Recognition, NUS CS4243, 2022

NaijaSenti is an open-source sentiment and emotion corpora for four major Nigerian languages

Advances in Neural Information Processing Systems (NeurIPS), 2020.

This project provides the proof of the uniqueness of the equilibrium and the global asymptotic stability.

PyTorch implementation of MoCo: Momentum Contrast for Unsupervised Visual Representation Learning

Trading Gym is an open source project for the development of reinforcement learning algorithms in the context of trading.

Myia prototyping

Shape-aware Semi-supervised 3D Semantic Segmentation for Medical Images

A Comprehensive Empirical Study of Vision-Language Pre-trained Model for Supervised Cross-Modal Retrieval

POCO: Point Convolution for Surface Reconstruction

A Python library for working with arbitrary-dimension hypercomplex numbers following the Cayley-Dickson construction of algebras.

Face Mask Detector by live camera using tensorflow-keras, openCV and Python

Dense Gaussian Processes for Few-Shot Segmentation

Deep Residual Learning for Image Recognition