Open Watson52 opened 2 years ago
strict=False
when loading the weights.pred_heatmap
is the predicted Gaussian "object centerness" map. It follows the CenterNet/CenterPoint [1,2] formulation [1] Objects as Points, Zhou et al., 2019 [2] Center-based 3D Object Detection and Tracking, Yin et al., 2021
Hi Chen, thanks for the wonderful work. I would like to ask you some questions about the code in
lav_agent.py
.bev_model
? Since I think thebev_planner
(teacher) only used to generate soft labels forUniPlanner
(student) to learn in the training time.pred_heatmaps
, but I think the corresponding output is the center location of other vehicles. Could you explain what aheatmap
is? https://github.com/dotchen/LAV/blob/2d00c6276d1a72f45627b4b9b418cd2ccad0b994/team_code/lav_agent.py#L220-L224 https://github.com/dotchen/LAV/blob/2d00c6276d1a72f45627b4b9b418cd2ccad0b994/lav/models/lidar.py#L34-L44