Skip to content

CenterNetLoss

CenterNetLoss class

dlf.losses.centernet_loss.CenterNetLoss(loss_scale=0.1, loss_offset=1.0)

Implementation of CenterNet loss

Aliases

  • CenternetLoss
  • centernetloss
  • center_net_loss
  • centernet_loss

Arguments

  • loss_scale: int. Scaling factor for the loss
  • loss_offset: float. Offset factor for the loss

YAML Configuration

loss:
    centernetloss:
        loss_scale: 0.1
        loss_offset: 1.0

References