Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/dcnv3.git" did not exist on "67ec718805d9648e280a94e323191a3bb7594a54"
Commit 59ed7304 authored by acivgin1's avatar acivgin1
Browse files

coords need to be contiguous

parent 19de178d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pcdet/models/roi_heads/partA2_head.py
View file @ 59ed7304
...@@ -192,7 +192,7 @@ class PartA2FCHead(RoIHeadTemplate): ...@@ -192,7 +192,7 @@ class PartA2FCHead(RoIHeadTemplate):
part_features = pooled_part_features[sparse_idx[:, 0], sparse_idx[:, 1], sparse_idx[:, 2], sparse_idx[:, 3]] part_features = pooled_part_features[sparse_idx[:, 0], sparse_idx[:, 1], sparse_idx[:, 2], sparse_idx[:, 3]]
rpn_features = pooled_rpn_features[sparse_idx[:, 0], sparse_idx[:, 1], sparse_idx[:, 2], sparse_idx[:, 3]] rpn_features = pooled_rpn_features[sparse_idx[:, 0], sparse_idx[:, 1], sparse_idx[:, 2], sparse_idx[:, 3]]
coords = sparse_idx.int() coords = sparse_idx.int().contiguous()
part_features = spconv.SparseConvTensor(part_features, coords, sparse_shape, batch_size_rcnn) part_features = spconv.SparseConvTensor(part_features, coords, sparse_shape, batch_size_rcnn)
rpn_features = spconv.SparseConvTensor(rpn_features, coords, sparse_shape, batch_size_rcnn) rpn_features = spconv.SparseConvTensor(rpn_features, coords, sparse_shape, batch_size_rcnn)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment