search

xiong-zhitong / pytorch_RFCN

279 stars 54 forks source link

ValueError: all the input array dimensions except for the concatenation axis must match exactly #3

Open BinWang-shu opened 7 years ago

BinWang-shu commented 7 years ago

I have the problem that: File "train.py", line 124, in net(im_data, im_info, gt_boxes, gt_ishard, dontcare_areas) File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py", line 206, in call result = self.forward(*input, **kwargs) File "/home/wangbin/wb/pytorch/pytorch_RFCN-master/faster_rcnn/rfcn.py", line 236, in forward roi_data = self.proposal_target_layer(rois, gt_boxes, gt_ishard, dontcare_areas, self.n_classes) File "/home/wangbin/wb/pytorch/pytorch_RFCN-master/faster_rcnn/rfcn.py", line 305, in proposal_target_layer proposal_target_layer_py(rpn_rois, gt_boxes, gt_ishard, dontcare_areas, num_classes) File "/home/wangbin/wb/pytorch/pytorch_RFCN-master/faster_rcnn/rpn_msr/proposal_target_layer.py", line 66, in proposal_target_layer np.hstack((zeros, np.vstack((gt_easyboxes[:, :-1], jittered_gt_boxes[:, :-1])))))) File "/usr/local/lib/python2.7/dist-packages/numpy/core/shape_base.py", line 230, in vstack return _nx.concatenate([atleast_2d(_m) for _m in tup], 0) ValueError: all the input array dimensions except for the concatenation axis must match exactly

I find that the "rcnn_cls" is "nan" and the features generated by VGG16 are zeros. Could you help me!

liyuanyaun commented 7 years ago

you can check your boxes annotation

hellooctopus commented 5 years ago

I meet the same questions. How did you solve it?