search

dvlab-research / FocalsConv

Focal Sparse Convolutional Networks for 3D Object Detection (CVPR 2022, Oral)
https://arxiv.org/abs/2204.12463
Apache License 2.0
368 stars 35 forks source link

ValueError: cannot reshape array #34

Open qweasd852 opened 6 months ago

qweasd852 commented 6 months ago

Traceback (most recent call last): File "/home/systemlab/FocalsConv/OpenPCDet/tools/train.py", line 201, in main() File "/home/systemlab/FocalsConv/OpenPCDet/tools/train.py", line 153, in main train_model( File "/home/systemlab/FocalsConv/OpenPCDet/tools/train_utils/train_utils.py", line 111, in train_model accumulated_iter = train_one_epoch( File "/home/systemlab/FocalsConv/OpenPCDet/tools/train_utils/train_utils.py", line 25, in train_one_epoch batch = next(dataloader_iter) File "/home/systemlab/anaconda3/envs/voxelrcnn/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 521, in next data = self._next_data() File "/home/systemlab/anaconda3/envs/voxelrcnn/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 561, in _next_data data = self._dataset_fetcher.fetch(index) # may raise StopIteration File "/home/systemlab/anaconda3/envs/voxelrcnn/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/systemlab/anaconda3/envs/voxelrcnn/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/systemlab/FocalsConv/OpenPCDet/tools/../pcdet/datasets/kitti/kitti_dataset.py", line 425, in getitem data_dict = self.prepare_data(data_dict=input_dict) File "/home/systemlab/FocalsConv/OpenPCDet/tools/../pcdet/datasets/dataset.py", line 130, in prepare_data data_dict = self.data_augmentor.forward( File "/home/systemlab/FocalsConv/OpenPCDet/tools/../pcdet/datasets/augmentor/data_augmentor.py", line 243, in forward data_dict = cur_augmentor(data_dict=data_dict) File "/home/systemlab/FocalsConv/OpenPCDet/tools/../pcdet/datasets/augmentor/database_sampler.py", line 419, in call data_dict = self.add_sampled_boxes_to_scene(data_dict, File "/home/systemlab/FocalsConv/OpenPCDet/tools/../pcdet/datasets/augmentor/database_sampler.py", line 264, in add_sampled_boxes_to_scene obj_points=obj_points.reshape([-1, self.sampler_cfg.NUM_POINT_FEATURES]) ValueError: cannot reshape array of size 942 into shape (4)

How can I solve this problem?