PointPillars

Link to the original research paper: https://arxiv.org/pdf/1812.05784.pdf

Investigate multihead? https://ieeexplore.ieee.org/document/9302666

Task 2

  • Change the lidar_cuda/Dockerfile
  • Change the params.h file
  • Change the lidar_cuda/launch file

So you need to load the models from here https://git.uwaterloo.ca/WATonomous/perception-models/-/tree/add_pp/pp, and then run it on the rosbag and see how it performs.

You already saw how alvin did it with pp_og

You need to change this file: https://git.uwaterloo.ca/WATonomous/wato_monorepo/-/merge_requests/292/diffs#e9343d054f95aaaaabc520e04d7ba50bba29f471 with the corresponding weights from here: https://git.uwaterloo.ca/WATonomous/perception-models/-/blob/add_pp/pp/pp_kitti_small_params.h

Run the model on a rosbag

Record and then share performance on Discord container name: lidar_cuda

year 3 is causing crashes,

/mnt/wato-drive/rosbags/year3/test_track_days
 
watod run bag_bridge rosbag play -r 1.0 /bags/year3/test_track_days/W20.0/public_road_to_TT.bag
python3 train.py --cfg_file /home/OpenPCDet/tools/cfgs/nuscenes_models/pointpillar_newaugs.yaml
python3 test.py --cfg_file /home/OpenPCDet/tools/cfgs/nuscenes_models/pointpillar_newaugs.yaml --ckpt /home/KITTI/ckpt --eval_all
 

To test if gpu supports the current torch version or whatever, run

import torch 
torch.zeros(1).cuda()
python setup.py develop
cd tools
python train.py --cfg_file cfgs/da-nuscenes-wato_models/pointpillar_newaugs.yaml