使用tf-Faster-RCNN进行对象检测时出现InvalidArgumentError

Question

使用tensorflows Faster-RCNN进行对象检测时，我遇到以下错误：

"InvalidArgumentError : ValueError: attempt to get argmax of an empty sequence  [[Node: model/rpn/target/PyFunc = PyFunc[Tin=[DT_FLOAT, DT_INT32, DT_INT32, 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"](model/rpn/cls/conv_1/Mul/_2579, _arg_Placeholder_2_0_2, _arg_Placeholder_1_0_1, model/rpn/target/PyFunc/input_3, model/rpn/target/PyFunc/input_4)]]
 [[Node: losses/fast_rcnn_cls_loss/SparseSoftmaxCrossEntropyWithLogits/Shape/_2565 = _HostRecv[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_5640_losses/fast_rcnn_cls_loss/SparseSoftmaxCrossEntropyWithLogits/Shape", tensor_type=DT_INT32, _device="/job:localhost/replica:0/task:0/gpu:0"]()]]".

我已检查过我的所有图片都包含有效的边框。我该如何解决这个问题？

Answer 1

TL; DR：从您的数据中移除零宽度边界框，也从the caches移除

当我收到此错误时，我正在使用基于Fast（er）-RCNN的CTPN文本检测器。

我注意到我的一些边界框有width=0。所以我清理了它们，但错误仍然存​​在。由于代码使用缓存，因此未反映更改。所以我从data文件夹中删除了缓存并重新运行代码。

希望这有助于某人。