15 lines (10 loc) · 524 Bytes

Train and Test

The training and testing is exactly the same as MapTRv2 with different configs

cd SemVecNet/SemVecNet

# For AV2
./tools/dist_train.sh projects/configs/semvecnet/semvecnet_av2_centerline.py gpu_num

# For nuScenes
./tools/dist_train.sh projects/configs/semvecnet/semvecnet_nusc_centerline.py gpu_num

The weights for SemVecNet can be found here.