Method __init__

detrsmpl/models/losses/iou_loss.py:318–323

(self, beta=0.2, eps=1e-3, reduction='mean', loss_weight=1.0)

class BoundedIoULoss(nn.Module):
    def __init__(self, beta=0.2, eps=1e-3, reduction='mean', loss_weight=1.0):
        super(BoundedIoULoss, self).__init__()
        self.beta = beta
        self.eps = eps
        self.reduction = reduction
        self.loss_weight = loss_weight

    def forward(self,
                pred,