When I started to train end-to-end , the error happened:
W tensorflow/core/framework/op_kernel.cc:1158] Unknown: KeyError: b'TRAIN'
Caused by op 'RPN/rpn_rois/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 427, in setup
.proposal_layer(_feat_stride[2:], anchor_size[2:], 'TRAIN',name = 'rpn_rois'))
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 345, in proposal_layer
[tf.float32]),\
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/ops/script_ops.py", line 198, in py_func
input=inp, token=token, Tout=Tout, name=name)
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/ops/gen_script_ops.py", line 38, in _py_func
name=name)
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 767, in apply_op
op_def=op_def)
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 2506, in create_op
original_op=self._default_original_op, op_def=op_def)
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1269, in init
self._traceback = _extract_stack()
yangyuke001
commented
6 years ago
When I started to train end-to-end , the error happened:
W tensorflow/core/framework/op_kernel.cc:1158] Unknown: KeyError: b'TRAIN'
Caused by op 'RPN/rpn_rois/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 427, in setup
.proposal_layer(_feat_stride[2:], anchor_size[2:], 'TRAIN',name = 'rpn_rois'))
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 345, in proposal_layer
[tf.float32]),\
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/ops/script_ops.py", line 198, in py_func
input=inp, token=token, Tout=Tout, name=name)
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/ops/gen_script_ops.py", line 38, in _py_func
name=name)
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 767, in apply_op
op_def=op_def)
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 2506, in create_op
original_op=self._default_original_op, op_def=op_def)
File "/home/gtx980a-206a/anaconda/envs/py3/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1269, in init
self._traceback = _extract_stack()
UnknownError (see above for traceback): KeyError: b'TRAIN' [[Node: RPN/rpn_rois/PyFunc = PyFunc[Tin=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_STRING, DT_INT32, DT_INT32], Tout=[DT_FLOAT], token="pyfunc_1", _device="/job:localhost/replica:0/task:0/cpu:0"](RPN/Reshape_2/_717, RPN/rpn_bbox_pred/BiasAdd/_719, RPN/Reshape_5/_721, RPN/rpn_bbox_pred_1/BiasAdd/_723, RPN/Reshape_8/_725, RPN/rpn_bbox_pred_2/BiasAdd/_727, RPN/Reshape_11/_729, RPN/rpn_bbox_pred_3/BiasAdd/_731, RPN/Reshape_14/_733, RPN/rpn_bbox_pred_4/BiasAdd/_735, _arg_im_info_0_5, RPN/rpn_rois/PyFunc/input_11, RPN/rpn_rois/PyFunc/input_12, RPN/rpn_rois/PyFunc/input_13)]] [[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"]()]]