Skip to content

Latest commit

 

History

History
40 lines (27 loc) · 838 Bytes

GETTING_STARTED.md

File metadata and controls

40 lines (27 loc) · 838 Bytes

Getting Started

Test and evaluate the pretrained models

  • Go to tools:
cd OpenPCDet/tools
  • Test with a pretrained model:
python test.py --cfg_file ${CONFIG_FILE} --batch_size ${BATCH_SIZE} --ckpt ${CKPT}
  • For example:
python test.py --cfg_file cfgs/kitti_models/pointpillar_fsa.yaml --batch_size 4 --ckpt ${SAVED_CKPT_PATH}/pointpillar_fsa.pth

Train a model

  • Train with multiple GPUs:
sh scripts/dist_train.sh ${NUM_GPUS} --cfg_file ${CONFIG_FILE} --batch_size ${BATCH_SIZE}  --epochs 80
  • For example:
sh scripts/dist_train.sh 4 --cfg_file cfgs/kitti_models/pointpillar_fsa.yaml  --epochs 80
  • Train with a single GPU:
python train.py --cfg_file ${CONFIG_FILE} --batch_size ${BATCH_SIZE} --epochs 50