训练速度更快的R-CNN时出错

Question

我正试着用我的电脑和一些同行在Ubuntu 16.04上运行更快的R-CNN。但是我被卡住了。我在谷歌上找不到和我一样的错误。

Normalizing targets
done
Solving...
Traceback (most recent call last):
  File "./tools/train_net.py", line 96, in <module>
    max_iters=args.max_iters)
  File "/home/david/Faster-RCNN_TF/tools/../lib/fast_rcnn/train.py", line 222, in train_net
    sw.train_model(sess, max_iters)
  File "/home/david/Faster-RCNN_TF/tools/../lib/fast_rcnn/train.py", line 95, in train_model
    rpn_cross_entropy = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(rpn_cls_score, rpn_label))
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/nn_ops.py", line 1686, in sparse_softmax_cross_entropy_with_logits
    labels, logits)
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/nn_ops.py", line 1535, in _ensure_xent_args
    "named arguments (labels=..., logits=..., ...)" % name)
ValueError: Only call `sparse_softmax_cross_entropy_with_logits` with named arguments (labels=..., logits=..., ...)

我不知道这是什么错误，也不知道如何解决这个错误。请帮帮我

Answer 1

而不是这样：

rpn_cross_entropy = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(rpn_cls_score, rpn_label))

试着写下：

tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(rpn_label, rpn_cls_score))

如果不起作用，请尝试：

tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(labels=rpn_label, logits=rpn_cls_score))