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

Question

我正尝试使用tensorflow.py_func将python函数'IoU'用作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/。