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

2021-AIAC-QQ-Browser-Hyperparameter-Optimization-Rank6

这是一个yolo3-tf2的源码，可以用于训练自己的模型。

This repository contains the code needed to train Mega-NeRF models and generate the sparse voxel octrees

Teaching end to end workflow of deep learning

Pytorch implementation of NeurIPS 2021 paper: Geometry Processing with Neural Fields.

ImageNet Adversarial Image Evaluation

Benchmark library for high-dimensional HPO of black-box models based on Weighted Lasso regression

A bare-bones Python library for quality diversity optimization.

[ICCV '21] In this repository you find the code to our paper Keypoint Communities

An updated version of virtual model making

Static Features Classifier - A static features classifier for Point-Could clusters using an Attention-RNN model

Range Image-based LiDAR Localization for Autonomous Vehicles Using Mesh Maps

Rest API Written In Python To Classify NSFW Images.

ThunderGBM: Fast GBDTs and Random Forests on GPUs

Pytorch implementation of the paper Improving Text-to-Image Synthesis Using Contrastive Learning

LexGLUE: A Benchmark Dataset for Legal Language Understanding in English

[ACM MM 2019 Oral] Cycle In Cycle Generative Adversarial Networks for Keypoint-Guided Image Generation

A Conditional Point Diffusion-Refinement Paradigm for 3D Point Cloud Completion

Official implementation of "SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformers"

Scalable Multi-Agent Reinforcement Learning