3DSSD-pytorch-openPCDet icon indicating copy to clipboard operation
3DSSD-pytorch-openPCDet copied to clipboard

Question about vote_loss

Open gordonyfg opened this issue 3 years ago • 0 comments

In tensorflow implementation and your 3DSSD-pytorch implementation, I can see there is a vote_loss. https://github.com/qiqihaer/3DSSD-pytorch/blob/d0c856e4c1b6cb69810fc0b0562ebc9a42e5bace/lib/builder/loss_builder.py#L190

Is it still available in this repo? If not, should I add vote_loss to improve the performance?

And I am not sure how to get end_points[maps_dict.PL_LABEL_BOXES_3D] from forward_ret_dict. Could you give me some hints please?

vote_offset = self.forward_ret_dict['ctr_offsets']
vote_base = self.forward_ret_dict['centers_origin']
bs, pts_num, _ = vote_offset.shape
gt_boxes_3d = end_points[maps_dict.PL_LABEL_BOXES_3D] 

Appreciate your efforts and thank you for your time

gordonyfg avatar Apr 27 '21 02:04 gordonyfg