BinaryPenaltyReducedFocalCrossEntropy
类keras_cv.losses.BinaryPenaltyReducedFocalCrossEntropy(
alpha=2.0,
beta=4.0,
from_logits=False,
positive_threshold=0.99,
positive_weight=1.0,
negative_weight=1.0,
reduction="sum_over_batch_size",
name="binary_penalty_reduced_focal_cross_entropy",
)
实现 CenterNet 修改后的焦点损失。
与 keras.losses.BinaryFocalCrossentropy
相比,此损失函数会对值小于 positive_threshold
的负标签进行折扣,负标签的值越大,对最终损失的折扣就越大。
用户可以选择在损失计算之外除以关键点的数量,或者通过将 sample_weight
设为 1.0/num_key_points 来实现。
参数
keras.losses.BinaryFocalCrossentropy
中的 gamma
参数。y_pred
是否应为 logits 张量,默认为 False
。输入:y_true:[batch_size, ...] 浮点型张量 y_pred:[batch_size, ...] 与 y_true 形状相同的浮点型张量。
参考文献
alpha
和 beta
。