All Projects → rayguan97 → M3DETR

rayguan97 / M3DETR

Licence: Apache-2.0 license
Code base for M3DeTR: Multi-representation, Multi-scale, Mutual-relation 3D Object Detection with Transformers

Programming Languages

python
139335 projects - #7 most used programming language
Cuda
1817 projects
C++
36643 projects - #6 most used programming language

Projects that are alternatives of or similar to M3DETR

efficient online learning
Efficient Online Transfer Learning for 3D Object Detection in Autonomous Driving
Stars: ✭ 20 (-57.45%)
Mutual labels:  point-cloud, kitti-dataset, 3d-object-detection
Awesome-3D-Object-Detection-for-Autonomous-Driving
Papers on 3D Object Detection for Autonomous Driving
Stars: ✭ 52 (+10.64%)
Mutual labels:  kitti-dataset, 3d-object-detection, waymo-open-dataset
ViP
A New 3D Detector. Code Will be made public.
Stars: ✭ 29 (-38.3%)
Mutual labels:  point-cloud, 3d-object-detection
Waymo Kitti Adapter
A tool converting Waymo dataset format to Kitti dataset format.
Stars: ✭ 83 (+76.6%)
Mutual labels:  kitti-dataset, waymo-open-dataset
BtcDet
Behind the Curtain: Learning Occluded Shapes for 3D Object Detection
Stars: ✭ 104 (+121.28%)
Mutual labels:  point-cloud, 3d-object-detection
progressive-coordinate-transforms
Progressive Coordinate Transforms for Monocular 3D Object Detection, NeurIPS 2021
Stars: ✭ 55 (+17.02%)
Mutual labels:  kitti-dataset, waymo-open-dataset
Overlappredator
[CVPR 2021, Oral] PREDATOR: Registration of 3D Point Clouds with Low Overlap.
Stars: ✭ 106 (+125.53%)
Mutual labels:  point-cloud, transformer
FLAT
[ICCV2021 Oral] Fooling LiDAR by Attacking GPS Trajectory
Stars: ✭ 52 (+10.64%)
Mutual labels:  point-cloud, 3d-object-detection
frustum-convnet
The PyTorch Implementation of F-ConvNet for 3D Object Detection
Stars: ✭ 228 (+385.11%)
Mutual labels:  point-cloud, 3d-object-detection
OverlapPredator
[CVPR 2021, Oral] PREDATOR: Registration of 3D Point Clouds with Low Overlap.
Stars: ✭ 293 (+523.4%)
Mutual labels:  point-cloud, transformer
torch-points3d
Pytorch framework for doing deep learning on point clouds.
Stars: ✭ 1,823 (+3778.72%)
Mutual labels:  point-cloud
deformer
[ACL 2020] DeFormer: Decomposing Pre-trained Transformers for Faster Question Answering
Stars: ✭ 111 (+136.17%)
Mutual labels:  transformer
point-cloud-prediction
Self-supervised Point Cloud Prediction Using 3D Spatio-temporal Convolutional Networks
Stars: ✭ 97 (+106.38%)
Mutual labels:  point-cloud
Transformer tf2.0
Transfromer tensorflow2.0版本实现
Stars: ✭ 23 (-51.06%)
Mutual labels:  transformer
CSV2RDF
Streaming, transforming, SPARQL-based CSV to RDF converter. Apache license.
Stars: ✭ 48 (+2.13%)
Mutual labels:  transformer
Walk-Transformer
From Random Walks to Transformer for Learning Node Embeddings (ECML-PKDD 2020) (In Pytorch and Tensorflow)
Stars: ✭ 26 (-44.68%)
Mutual labels:  transformer
KittiViz
Data visualizer for Kitti dataset.
Stars: ✭ 18 (-61.7%)
Mutual labels:  kitti-dataset
transformer-models
Deep Learning Transformer models in MATLAB
Stars: ✭ 90 (+91.49%)
Mutual labels:  transformer
text-style-transfer-benchmark
Text style transfer benchmark
Stars: ✭ 56 (+19.15%)
Mutual labels:  transformer
RealSense
Extension of RealSense Unity Wrapper [Unofficial]
Stars: ✭ 31 (-34.04%)
Mutual labels:  point-cloud

M3DETR: Multi-representation, Multi-scale, Mutual-relation 3D Object Detection with Transformers

The code base for M3DETR: Multi-representation, Multi-scale, Mutual-relation 3D Object Detection with Transformers
Tianrui Guan*, Jun Wang*, Shiyi Lan, Rohan Chandra, Zuxuan Wu, Larry Davis, Dinesh Manocha

Abstract

We present a novel architecture for 3D object detection, M3DETR, which combines different point cloud representations (raw, voxels, bird-eye view) with different feature scales based on multi-scale feature pyramids. M3DETR is the first approach that unifies multiple point cloud representations, feature scales, as well as models mutual relationships between point clouds simultaneously using transformers. We perform extensive ablation experiments that highlight the benefits of fusing representation and scale, and modeling the relationships. Our method achieves state-of-the-art performance on the KITTI 3D object detection dataset and Waymo Open Dataset. Results show that M3DETR improves the baseline significantly by 1.48% mAP for all classes on Waymo Open Dataset. In particular, our approach ranks 1st on the well-known KITTI 3D Detection Benchmark for both car and cyclist classes, and ranks 1st on Waymo Open Dataset with single frame point cloud input.

 

Features

  • A unified architecture for 3D object detection with transformers that accounts for multi-representation, multi-scale, mutual-relation models of point clouds in an end-to-end manner.
  • Support major 3D object detection datasets: Waymo Open Dataset, KITTI.

Installation

See installation instructions.

Getting Started

See Getting Started with M3DETR.

Model Zoo and Baselines

We provide a large set of baseline results and trained models available for download in the M3DETR Model Zoo.

Citation

Please cite our work if you found it useful,

@InProceedings{Guan_2022_WACV,
    author    = {Guan, Tianrui and Wang, Jun and Lan, Shiyi and Chandra, Rohan and Wu, Zuxuan and Davis, Larry and Manocha, Dinesh},
    title     = {M3DETR: Multi-Representation, Multi-Scale, Mutual-Relation 3D Object Detection With Transformers},
    booktitle = {Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)},
    month     = {January},
    year      = {2022},
    pages     = {772-782}
}

License

This project is released under the Apache 2.0 license.

Acknowledgement

The source code of M3DETR is based on OpenPCDet.

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].