xmyqsh / FPN

Feature Pyramid Network
155 stars 58 forks source link

ValueError: attempt to get argmax of an empty sequence #21

Open RoseLii opened 6 years ago

RoseLii commented 6 years ago

please help me .when i train my own data 2017-11-15 21:07:00.175803: W tensorflow/core/framework/op_kernel.cc:1158] Invalid argument: exceptions.ValueError: attempt to get argmax of an empty sequence [[Node: RPN/rpn-data/PyFunc = PyFunc[Tin=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_INT32, DT_FLOAT, DT_FLOAT, DT_INT32, DT_INT32], Tout=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT], token="pyfunc_0", _device="/job:localhost/replica:0/task:0/cpu:0"](RPN/rpn_cls_score/BiasAdd/_587, RPN/rpn_cls_score_1/BiasAdd/_589, RPN/rpn_cls_score_2/BiasAdd/_591, RPN/rpn_cls_score_3/BiasAdd/_593, RPN/rpn_cls_score_4/BiasAdd/_595, _arg_gt_boxes_0_3, _arg_gt_ishard_0_4, _arg_dontcare_areas_0_2, _arg_im_info_0_5, RPN/rpn-data/PyFunc/input_9, RPN/rpn-data/PyFunc/input_10)]] Traceback (most recent call last): File "./faster_rcnn/train_net.py", line 109, in restore=bool(int(args.restore))) File "./faster_rcnn/../lib/fast_rcnn/train.py", line 409, in train_net sw.train_model(sess, max_iters, restore=restore) File "./faster_rcnn/../lib/fast_rcnn/train.py", line 263, in train_model cls_prob, bbox_pred, rois = sess.run(fetches=fetch_list, feed_dict=feed_dict) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 789, in run run_metadata_ptr) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 997, in _run feed_dict_string, options, run_metadata) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 1132, in _do_run target_list, options, run_metadata) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 1152, in _do_call raise type(e)(node_def, op, message) tensorflow.python.framework.errors_impl.InvalidArgumentError: exceptions.ValueError: attempt to get argmax of an empty sequence [[Node: RPN/rpn-data/PyFunc = PyFunc[Tin=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_INT32, DT_FLOAT, DT_FLOAT, DT_INT32, DT_INT32], Tout=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT], token="pyfunc_0", _device="/job:localhost/replica:0/task:0/cpu:0"](RPN/rpn_cls_score/BiasAdd/_587, RPN/rpn_cls_score_1/BiasAdd/_589, RPN/rpn_cls_score_2/BiasAdd/_591, RPN/rpn_cls_score_3/BiasAdd/_593, RPN/rpn_cls_score_4/BiasAdd/_595, _arg_gt_boxes_0_3, _arg_gt_ishard_0_4, _arg_dontcare_areas_0_2, _arg_im_info_0_5, RPN/rpn-data/PyFunc/input_9, RPN/rpn-data/PyFunc/input_10)]] [[Node: gradients/RPN/rpn_bbox_pred_reshape_concat_grad/Gather_13/_619 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/gpu:0", send_device="/job:localhost/replica:0/task:0/cpu:0", send_device_incarnation=1, tensor_name="edge_2711_gradients/RPN/rpn_bbox_pred_reshape_concat_grad/Gather_13", tensor_type=DT_INT64, _device="/job:localhost/replica:0/task:0/gpu:0"]()]]

Caused by op u'RPN/rpn-data/PyFunc', defined at: File "./faster_rcnn/train_net.py", line 101, in network = get_network(args.network_name) File "./faster_rcnn/../lib/networks/factory.py", line 22, in get_network return FPN_train() File "./faster_rcnn/../lib/networks/FPN_train.py", line 25, in init self.setup() File "./faster_rcnn/../lib/networks/FPN_train.py", line 418, in setup .anchor_target_layer(_feat_stride[2:], anchor_size[2:], name = 'rpn-data')) File "./faster_rcnn/../lib/networks/network.py", line 34, in layer_decorated layer_output = op(self, layer_input, *args, **kwargs) File "./faster_rcnn/../lib/networks/network.py", line 380, in anchor_target_layer [tf.float32,tf.float32,tf.float32,tf.float32]) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/script_ops.py", line 198, in py_func input=inp, token=token, Tout=Tout, name=name) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/gen_script_ops.py", line 38, in _py_func name=name) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/op_def_library.py", line 767, in apply_op op_def=op_def) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 2506, in create_op original_op=self._default_original_op, op_def=op_def) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 1269, in init self._traceback = _extract_stack()

InvalidArgumentError (see above for traceback): exceptions.ValueError: attempt to get argmax of an empty sequence [[Node: RPN/rpn-data/PyFunc = PyFunc[Tin=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_INT32, DT_FLOAT, DT_FLOAT, DT_INT32, DT_INT32], Tout=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT], token="pyfunc_0", _device="/job:localhost/replica:0/task:0/cpu:0"](RPN/rpn_cls_score/BiasAdd/_587, RPN/rpn_cls_score_1/BiasAdd/_589, RPN/rpn_cls_score_2/BiasAdd/_591, RPN/rpn_cls_score_3/BiasAdd/_593, RPN/rpn_cls_score_4/BiasAdd/_595, _arg_gt_boxes_0_3, _arg_gt_ishard_0_4, _arg_dontcare_areas_0_2, _arg_im_info_0_5, RPN/rpn-data/PyFunc/input_9, RPN/rpn-data/PyFunc/input_10)]] [[Node: gradients/RPN/rpn_bbox_pred_reshape_concat_grad/Gather_13/_619 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/gpu:0", send_device="/job:localhost/replica:0/task:0/cpu:0", send_device_incarnation=1, tensor_name="edge_2711_gradients/RPN/rpn_bbox_pred_reshape_concat_grad/Gather_13", tensor_type=DT_INT64, _device="/job:localhost/replica:0/task:0/gpu:0"]()]]