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
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