HKUST-Aerial-Robotics / Stereo-RCNN

Code for 'Stereo R-CNN based 3D Object Detection for Autonomous Driving' (CVPR 2019)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

when I running test.sh ValueError

WUT-xiaoming opened this issue · comments

Loaded dataset kitti_val for training
Preparing training data...
cache file ./data/kitti/object/kitti_val_gt_roidb.pkl
kitti_val gt roidb loaded from ./data/kitti/object/kitti_val_gt_roidb.pkl
done
3769 roidb entries
load checkpoint models_stereo/stereo_rcnn_12_6477.pth
load model successfully!
Traceback (most recent call last):s, solve time 0.39s (Press Esc to exit!)
File "test_net.py", line 121, in
data = next(data_iter)
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 615, in next
batch = self.collate_fn([self.dataset[i] for i in indices])
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 615, in
batch = self.collate_fn([self.dataset[i] for i in indices])
File "/dataVolume/szm/Stereo-RCNN-1.0/lib/roi_data_layer/roibatchLoader.py", line 45, in getitem
blobs = get_minibatch(minibatch_db, self._num_classes)
File "/dataVolume/szm/Stereo-RCNN-1.0/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 "/dataVolume/szm/Stereo-RCNN-1.0/lib/roi_data_layer/minibatch.py", line 115, in _get_image_blob
blob_left, blob_right = im_list_to_blob(processed_ims_left, processed_ims_right)
File "/dataVolume/szm/Stereo-RCNN-1.0/lib/model/utils/blob.py", line 35, in im_list_to_blob
blob_right[i, 0:im_r.shape[0], 0:im_r.shape[1], :] = im_r
ValueError: could not broadcast input array from shape (608,2014,3) into shape (600,1985,3)