From 3d8f9f16aee0cf4ebf22e12430f14b488979fcb2 Mon Sep 17 00:00:00 2001 From: Caleb Robinson Date: Tue, 28 Dec 2021 13:36:06 -0800 Subject: [PATCH] Quick fix for losses docs (#333) * Fix doc link * Brute forcing rst syntax * Adding more docs to ChesapeakeCVPR --- torchgeo/datasets/chesapeake.py | 6 ++++++ torchgeo/losses/qr.py | 8 ++++---- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/torchgeo/datasets/chesapeake.py b/torchgeo/datasets/chesapeake.py index d8a2405681f..4c46b9fed91 100644 --- a/torchgeo/datasets/chesapeake.py +++ b/torchgeo/datasets/chesapeake.py @@ -288,6 +288,12 @@ class ChesapeakeCVPR(GeoDataset): This dataset was organized to accompany the 2019 CVPR paper, "Large Scale High-Resolution Land Cover Mapping with Multi-Resolution Data". + The paper "Resolving label uncertainty with implicit generative models" added an + additional layer of data to this dataset containing a prior over the Chesapeake Bay + land cover classes generated from the NLCD land cover labels. For more information + about this layer see `the dataset documentation + `_. + If you use this dataset in your research, please cite the following paper: * https://doi.org/10.1109/cvpr.2019.01301 diff --git a/torchgeo/losses/qr.py b/torchgeo/losses/qr.py index c6120d34fdf..e24d0d773cf 100644 --- a/torchgeo/losses/qr.py +++ b/torchgeo/losses/qr.py @@ -14,8 +14,8 @@ class QRLoss(Module): """The QR (forward) loss between class probabilities and predictions. - This loss is defined in 'Resolving label uncertainty with implicit generative - models' https://openreview.net/forum?id=AEa_UepnMDX. + This loss is defined in `'Resolving label uncertainty with implicit generative + models' `_. .. versionadded:: 0.2 """ @@ -45,8 +45,8 @@ def forward(self, probs: torch.Tensor, target: torch.Tensor) -> torch.Tensor: class RQLoss(Module): """The RQ (backwards) loss between class probabilities and predictions. - This loss is defined in 'Resolving label uncertainty with implicit generative - models' https://openreview.net/forum?id=AEa_UepnMDX. + This loss is defined in `'Resolving label uncertainty with implicit generative + models' `_. .. versionadded:: 0.2 """