README.md

DETR3D

This repo is to build a lightweight DETR3D without relying on MMDET3D.

Prepare Dataset

Follow MMDET3D to prepare dataset.

Current supported datasets: nuScenes

Download Pretrained Model

Downloads the pretrained backbone weights fcos3d to ckpts/.

Train

With mmdet3d

python tools/train.py configs/detr3d/detr3d_res101_gridmask.py

Without mmdet3d

python tools/train_wo_mmdet3d.py configs/detr3d/detr3d_res101_gridmask.py