yangJirui / RRPN_FPN_Tensorflow

84 stars 23 forks source link

Issue with libs/rpn/build_rpn.py during inference #4

Open lillythomas opened 6 years ago

lillythomas commented 6 years ago

Hi there,

Thank you for providing this repo. When I execute the inference script, I end up with the error ValueError: Dimension 0 in both shapes must be equal, but are 5 and 4 for 'rpn_proposals/cond_1/concat' (op: 'ConcatV2') with input shapes: [?,5], [?,4], [] and with computed input tensors: input[2] = <0>.

Full traceback: ++--++--++--++--++--++--++--++--++--++--++--++--++--++--++--++--++--++--++--++-- /mnt/cirrus/models/RRPN_FPN_Tensorflow Traceback (most recent call last): File "inference.py", line 165, in <module> inference() File "inference.py", line 87, in inference rpn_proposals_boxes, rpn_proposals_scores = rpn.rpn_proposals() # rpn_score shape: [300, ] File "../libs/rpn/build_rpn.py", line 468, in rpn_proposals lambda: (valid_boxes, valid_scores)) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/util/deprecation.py", line 289, in new_func return func(*args, **kwargs) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/ops/control_flow_ops.py", line 1814, in cond orig_res_t, res_t = context_t.BuildCondBranch(true_fn) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/ops/control_flow_ops.py", line 1689, in BuildCondBranch original_result = fn() File "../libs/rpn/build_rpn.py", line 465, in <lambda> self.max_proposals_num), File "../libs/box_utils/boxes_utils.py", line 92, in padd_boxes_with_zeros final_boxes = tf.concat([boxes, zero_boxes], axis=0) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/ops/array_ops.py", line 1048, in concat name=name) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/ops/gen_array_ops.py", line 495, in _concat_v2 name=name) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/framework/op_def_library.py", line 767, in apply_op op_def=op_def) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 2508, in create_op set_shapes_for_outputs(ret) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 1873, in set_shapes_for_outputs shapes = shape_func(op) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 1823, in call_with_requiring return call_cpp_shape_fn(op, require_shape_fn=True) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/framework/common_shapes.py", line 610, in call_cpp_shape_fn debug_python_shape_fn, require_shape_fn) File "/mnt/cirrus/venvs/python_27/local/lib/python2.7/site-packages/tensorflow/python/framework/common_shapes.py", line 676, in _call_cpp_shape_fn_impl raise ValueError(err.message) ValueError: Dimension 0 in both shapes must be equal, but are 5 and 4 for 'rpn_proposals/cond_1/concat' (op: 'ConcatV2') with input shapes: [?,5], [?,4], [] and with computed input tensors: input[2] = <0>.

I am running ubuntu 16.04, cuda 8.0.61, python 2.7, and tensorflow 1.2.0

If you could help me out here, that would be greatly appreciated! Thank you in advance.