在 keras 损失函数中使用 tensorflow.py_func

Question

我正在尝试使用 python 函数“IoU”作为使用 tensorflow.py_func 的 keras 损失函数，如下所示：

def loss_fn(label, pred):
    return tensorflow.py_func(IoU, (label, pred), tf.float64)

但不幸的是，我收到以下错误：

TypeError: 'NoneType' object cannot be interpreted as an integer.

我正在使用 Keras 2.2.2 和 tensorflow 1.10.0。

你能帮我解决这个问题吗？

提前致谢。

Answer 1

您只需要导入后端，然后您就可以直接传递 tensorflow 函数。有关详细信息，请参阅https://keras.io/activations/。