open3d.ml.tf.modules.losses.FocalLoss

class open3d.ml.tf.modules.losses.FocalLoss(gamma=2.0, alpha=0.25, loss_weight=1.0)

Focal Loss

Parameters
  • gamma (float, optional) – The gamma for calculating the modulating factor. Defaults to 2.0.

  • alpha (float, optional) – A balanced form for Focal Loss. Defaults to 0.25.

  • loss_weight (float, optional) – Weight of loss. Defaults to 1.0.

__init__(gamma=2.0, alpha=0.25, loss_weight=1.0)

Initialize self. See help(type(self)) for accurate signature.