Home

Awesome

MTrans

The PyTorch implementation of Multimodal Transformer for Automatic 3D Annotation and Object Detection, which has been accepted by ECCV2022.

Installation

The code has been tested on PyTorch v1.9.1.

IoU loss is required for training. Before running, please install the IoU loss package following this doc.

Data Preparation

The KITTI 3D detection dataset can be downloaded from the official webstie: link.

Train

To train a MTrans with the KITTI dataset. Simply run:

python train.py --cfg_file configs/MTrans_kitti.yaml

Trained Model

Trained checkpoint can be downloaded from here. Although we try to fix the random seeds, due to the randomness in some asynchronuous CUDA opearations and data preprocessing (e.g., point sampling), the result might not be exactly the same from run to run.

References

The IoU loss module is borrowed from "https://github.com/lilanxiao/Rotated_IoU". We thank the author for providing a neat implementation of the IoU loss.