search

HKUST-Aerial-Robotics / Stereo-RCNN

Code for 'Stereo R-CNN based 3D Object Detection for Autonomous Driving' (CVPR 2019)
MIT License
690 stars 177 forks source link

IndexError: too many indices for array #26

Open liluoqinchuan opened 5 years ago

liluoqinchuan commented 5 years ago

Hi PeiLiang: When I trained the model to 3682 iterations of the first epoch, I encountered the following problems:

IndexError: Traceback (most recent call last): File "/home/zhaodan/Downloads/sofaware/pycharm-community-2018.3.1/venvpy2/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 42, in _worker_loop samples = collate_fn([dataset[i] for i in batch_indices]) File "/home/zhaodan/Downloads/code/Stereo-RCNN-master1/lib/roi_data_layer/roibatchLoader.py", line 65, in getitem blobs = get_minibatch(minibatch_db, self._num_classes) File "/home/zhaodan/Downloads/code/Stereo-RCNN-master1/lib/roi_data_layer/minibatch.py", line 34, in get_minibatch im_blob_left, im_blob_right, im_scales = _get_image_blob(roidb, random_scale_inds) File "/home/zhaodan/Downloads/code/Stereo-RCNN-master1/lib/roi_data_layer/minibatch.py", line 114, in _get_image_blob img_right = img_right[:,:,::-1] IndexError: too many indices for array

Do you have such a problem?

My GPU is nvidia GTX 1081Ti , its memory size is 11G , does it enough?

Looking forward to your reply！ Thanks!

PeiliangLi commented 5 years ago

Are you using gray scale images?

liluoqinchuan commented 5 years ago

I downloaded the data according to the link given：

image

image

PeiliangLi commented 5 years ago

Can you print the shape of the img_right?