LittlePey / SFD

Sparse Fuse Dense: Towards High Quality 3D Detection with Depth Completion (CVPR 2022, Oral)
Apache License 2.0
263 stars 35 forks source link

what is the runtimeerror means, I 've already develop the sort_vertice #50

Closed kenfishball closed 1 year ago

kenfishball commented 1 year ago

屏幕截图 2023-03-22 144232 Traceback (most recent call last): File "train.py", line 201, in main() File "train.py", line 156, in main train_model( File "/mnt/data/zhuai/3D Point cloud/SFD/tools/train_utils/train_utils.py", line 86, in train_model accumulated_iter = train_one_epoch( File "/mnt/data/zhuai/3D Point cloud/SFD/tools/train_utils/train_utils.py", line 38, in train_one_epoch loss, tb_dict, disp_dict = model_func(model, batch) File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/models/init.py", line 30, in model_func ret_dict, tb_dict, disp_dict = model(batch_dict) File "/home/zhuai/miniconda3/envs/sfd/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl return forward_call(*input, **kwargs) File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/models/detectors/sfd.py", line 14, in forward loss, tb_dict, disp_dict = self.get_training_loss() File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/models/detectors/sfd.py", line 35, in get_training_loss loss_rcnn, tb_dict = self.roi_head.get_loss(tb_dict) File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/models/roi_heads/sfd_head.py", line 651, in get_loss rcnn_loss_reg, reg_tb_dict, iou_target = self.get_box_reg_layer_loss(self.forward_ret_dict) File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/models/roi_heads/sfd_head.py", line 742, in get_box_reg_layer_loss iou_loss, iou3d = oriented_iou_loss.cal_giou_3d(dt_boxes.unsqueeze(0), gt_boxes.unsqueeze(0)) File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/ops/iou3d/oriented_iou_loss.py", line 119, in cal_giou_3d iou3d, corners1, corners2, z_range, u3d = cal_iou_3d(box3d1, box3d2, verbose=True) File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/ops/iou3d/oriented_iou_loss.py", line 96, in cal_iou_3d iou_2d, corners1, corners2, u = cal_iou(box1, box2) # (B, N) File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/ops/iou3d/oriented_iou_loss.py", line 52, in cal_iou interarea, = oriented_box_intersection_2d(corners1, corners2) #(B, N) File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/ops/iou3d/box_intersection_2d.py", line 175, in oriented_box_intersection_2d sorted_indices = sort_indices(vertices, mask) File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/ops/iou3d/box_intersection_2d.py", line 141, in sort_indices return sort_v(vertices_normalized, mask, num_valid).long() File "/mnt/data/zhuai/3D Point cloud/SFD/pcdet/ops/iou3d/cuda_op/cuda_ext.py", line 9, in forward idx = sort_vertices.sort_vertices_forward(vertices, mask, num_valid) RuntimeError