**If you want to quickly develop your own model based on MPPNet, our recommended setting is to use mppnet_4frames.yaml, disable `USE_ROI_AUG` and `USE_TRAJ_AUG` flags in yaml and train 3 epoch. A reference time cost for this setting is about 5 hours, using 8 A100 GPUs. After finishing your development, you can get stable gains when using mppnet_16frames.yaml, enabling `USE_ROI_AUG` and `USE_TRAJ_AUG` flags and training 6 epoch.**
## Installation
## Installation
Please refer to [INSTALL.md](docs/INSTALL.md) for the installation of `OpenPCDet`.
Please refer to [INSTALL.md](docs/INSTALL.md) for the installation of `OpenPCDet`.
## Data Preparation
##Data Preparation
Please refer to [GETTING_STARTED.md](docs/GETTING_STARTED.md) to process the Waymo Open Dataset.
Please refer to [GETTING_STARTED.md](docs/GETTING_STARTED.md) to process the Waymo Open Dataset.
##Training
##Training
1. Train the RPN model for MPPNet (centerpoint_4frames is employed in the paper)
1. Train the RPN model for MPPNet (centerpoint_4frames is employed in the paper)
```shell
```shell
...
@@ -33,13 +35,14 @@ The prediction results of train and val dataset will be saved in
...
@@ -33,13 +35,14 @@ The prediction results of train and val dataset will be saved in
The default parameters in mppnet_e2e_memorybank_inference.yaml is for 4-frame and just change them to the setting in mppnet_16frames.yaml when using 16-frame.
The default parameters in mppnet_e2e_memorybank_inference.yaml is for 4-frame and just change them to the setting in mppnet_16frames.yaml when using 16-frame.