Skip to content

IndexError: too many indices for tensor of dimension 1 #1740

@TianchaoHuo

Description

@TianchaoHuo

Any suggestion to address the following issue? same as #1634

Traceback (most recent call last):
File "/home/user/admin/OpenPCDet/tools/demo.py", line 112, in
main()
File "/home/user/admin/OpenPCDet/tools/demo.py", line 98, in main
pred_dicts, _ = model.forward(data_dict)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/user/admin/OpenPCDet/pcdet/models/detectors/pointpillar.py", line 11, in forward
batch_dict = cur_module(batch_dict)
^^^^^^^^^^^^^^^^^^^^^^
File "/home/user/anaconda3/envs/pcdet/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/user/anaconda3/envs/pcdet/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/user/admin/OpenPCDet/pcdet/models/backbones_2d/map_to_bev/pointpillar_scatter.py", line 29, in forward
pillars = pillar_features[batch_mask, :]
~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^
IndexError: too many indices for tensor of dimension 1

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions