torchgeo.losses¶
TorchGeo losses.
- class torchgeo.losses.QRLoss(eps=1e-08)[source]¶
Bases:
Module
The QR (forward) loss between class probabilities and predictions.
This loss is defined in ‘Resolving label uncertainty with implicit generative models’.
New in version 0.2.
- __init__(eps=1e-08)[source]¶
Initialize a new QRLoss instance.
- Parameters:
eps (float) – small constant for numerical stability to prevent log(0) when computing the loss. Must be greater than or equal to 0.
- Raises:
ValueError – If eps is less than 0.
New in version 0.8: The eps parameter.
- class torchgeo.losses.RQLoss(eps=1e-08)[source]¶
Bases:
Module
The RQ (backwards) loss between class probabilities and predictions.
This loss is defined in ‘Resolving label uncertainty with implicit generative models’.
New in version 0.2.
- __init__(eps=1e-08)[source]¶
Initialize a new RQLoss instance.
- Parameters:
eps (float) – small constant for numerical stability to prevent division by zero and log(0) when computing the loss. Must be greater than or equal to 0.
- Raises:
ValueError – If eps is less than 0.