search

YuwenXiong / py-R-FCN

R-FCN with joint training and python support
MIT License
1.05k stars 472 forks source link

when i train rfcn_end2end,there is an error in proposal_target_layer.py #120

Open zhaowujie opened 5 years ago

zhaowujie commented 5 years ago

File "/media/hp208/4t/yangxueyuan/R-FCN/py-R-FCN/tools/train_net.py", line 112, in max_iters=args.max_iters) File "/media/hp208/4t/yangxueyuan/R-FCN/py-R-FCN/tools/../lib/fast_rcnn/train.py", line 205, in train_net model_paths = sw.train_model(max_iters) File "/media/hp208/4t/yangxueyuan/R-FCN/py-R-FCN/tools/../lib/fast_rcnn/train.py", line 146, in train_model self.solver.step(1) File "/media/hp208/4t/yangxueyuan/R-FCN/py-R-FCN/tools/../lib/rpn/proposal_target_layer.py", line 51, in forward (all_rois, np.hstack((zeros, gt_boxes[:, :-1]))) File "/usr/local/lib/python2.7/dist-packages/numpy/core/shape_base.py", line 288, in hstack return _nx.concatenate(arrs, 1) ValueError: all the input arrays must have same number of dimensions