Y_True 损失函数不正确

Y_True in Loss Function not Correct

我正在构建我的损失函数。但是,当打印 y_true 张量的值时,它打印的是带小数点的值(即 0.25、0.569、0.958)。这不应该是真的,因为 y_true 应该只有两个 类 0 或 1。这是我的代码:

@tf.function
def weighted_binary_crossentropy(y_true, y_pred):
    y_true= K.reshape(y_true, (K.shape(y_true)[0], -1))
    tf.print("tensors1:", y_true, output_stream=sys.stdout, summarize=50000)

为什么我得到这样的输出而不是 0 和 1?

我能够检测到来自我的数据生成器的问题。当我添加 rotation_range 时,它会弄乱像素值。这是因为在旋转时会创建新像素