MaskedSmoothL1Loss

class mmocr.models.common.MaskedSmoothL1Loss(beta=1, eps=1e-06)

Masked Smooth L1 loss.

参数

• beta (float, optional) – The threshold in the piecewise function. Defaults to 1.

• eps (float, optional) – Eps to avoid zero-division error. Defaults to 1e-6.

返回类型

None

forward(pred, gt, mask=None)

Forward function.

参数

• pred (torch.Tensor) – The prediction in any shape.

• gt (torch.Tensor) – The learning target of the prediction in the same shape as pred.

• mask (torch.Tensor, optional) – Binary mask in the same shape of pred, indicating positive regions to calculate the loss. Whole region will be taken into account if not provided. Defaults to None.

返回

The loss value.

返回类型

torch.Tensor