search

DetectionTeamUCAS / RRPN_Faster-RCNN_Tensorflow

A tensorflow re-implementation of RRPN: Arbitrary-Oriented Scene Text Detection via Rotation Proposals.
254 stars 89 forks source link

An confused error #17

Closed tsing-cv closed 5 years ago

tsing-cv commented 5 years ago 
  File "/home/aikaka/anaconda3/envs/tf_py36/lib/python3.6/site-packages/tensorflow/python/ops/script_ops.py", line 158, in __call__
    ret = func(*args)

  File "../libs/detection_oprations/proposal_target_layer.py", line 29, in proposal_target_layer
    fg_rois_per_image, rois_per_image, cfgs.CLASS_NUM+1)

  File "../libs/detection_oprations/proposal_target_layer.py", line 130, in _sample_rois
    bbox_targets = _get_bbox_regression_labels(bbox_target_data, num_classes)

  File "../libs/detection_oprations/proposal_target_layer.py", line 59, in _get_bbox_regression_labels
    bbox_targets[ind, start:end] = bbox_target_data[ind, 1:]

ValueError: could not broadcast input array from shape (5) into shape (0)

     [[Node: sample_RCNN_minibatch/PyFunc_1 = PyFunc[Tin=[DT_FLOAT, DT_FLOAT, DT_FLOAT], Tout=[DT_FLOAT, DT_FLOAT, DT_FLOAT], token="pyfunc_10", _device="/job:localhost/replica:0/task:0/device:CPU:0"](postprocess_RPN/GatherV2_1/_1257, get_batch/Reshape_2/_1181, sample_RCNN_minibatch/Reshape/_1265)]]
     [[Node: Fast-RCNN/rois_pooling/PyFunc/_1677 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device_incarnation=1, tensor_name="edge_3003_Fast-RCNN/rois_pooling/PyFunc", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"]()]]

@yangxue0827