diff --git a/docs/api/landsat_pretrained_weights.csv b/docs/api/landsat_pretrained_weights.csv index b62fd9f8be4..22493c4f368 100644 --- a/docs/api/landsat_pretrained_weights.csv +++ b/docs/api/landsat_pretrained_weights.csv @@ -1,32 +1,32 @@ Weight,Landsat,Channels,Source,Citation,License,NLCD (Acc),NLCD (mIoU),CDL (Acc),CDL (mIoU) -ResNet18_Weights.LANDSAT_TM_TOA_MOCO,4--5,5,`link `__,`link `__,"CC0-1.0",67.65,51.11,68.70,52.32 -ResNet18_Weights.LANDSAT_TM_TOA_SIMCLR,4--5,5,`link `__,`link `__,"CC0-1.0",60.86,43.74,61.94,44.86 -ResNet50_Weights.LANDSAT_TM_TOA_MOCO,4--5,5,`link `__,`link `__,"CC0-1.0",68.75,53.28,69.45,53.20 -ResNet50_Weights.LANDSAT_TM_TOA_SIMCLR,4--5,5,`link `__,`link `__,"CC0-1.0",62.05,44.98,62.80,45.77 -ViTSmall16_Weights.LANDSAT_TM_TOA_MOCO,4--5,5,`link `__,`link `__,"CC0-1.0",67.17,50.57,67.60,51.07 -ViTSmall16_Weights.LANDSAT_TM_TOA_SIMCLR,4--5,5,`link `__,`link `__,"CC0-1.0",66.82,50.17,66.92,50.28 -ResNet18_Weights.LANDSAT_ETM_TOA_MOCO,7,9,`link `__,`link `__,"CC0-1.0",65.22,48.39,62.84,45.81 -ResNet18_Weights.LANDSAT_ETM_TOA_SIMCLR,7,9,`link `__,`link `__,"CC0-1.0",58.76,41.60,56.47,39.34 -ResNet50_Weights.LANDSAT_ETM_TOA_MOCO,7,9,`link `__,`link `__,"CC0-1.0",66.60,49.92,64.12,47.19 -ResNet50_Weights.LANDSAT_ETM_TOA_SIMCLR,7,9,`link `__,`link `__,"CC0-1.0",57.17,40.02,54.95,37.88 -ViTSmall16_Weights.LANDSAT_ETM_TOA_MOCO,7,9,`link `__,`link `__,"CC0-1.0",63.75,46.79,60.88,43.70 -ViTSmall16_Weights.LANDSAT_ETM_TOA_SIMCLR,7,9,`link `__,`link `__,"CC0-1.0",63.33,46.34,59.06,41.91 -ResNet18_Weights.LANDSAT_ETM_SR_MOCO,7,6,`link `__,`link `__,"CC0-1.0",64.18,47.25,67.30,50.71 -ResNet18_Weights.LANDSAT_ETM_SR_SIMCLR,7,6,`link `__,`link `__,"CC0-1.0",57.26,40.11,54.42,37.48 -ResNet50_Weights.LANDSAT_ETM_SR_MOCO,7,6,`link `__,`link `__,"CC0-1.0",64.37,47.46,62.35,45.30 -ResNet50_Weights.LANDSAT_ETM_SR_SIMCLR,7,6,`link `__,`link `__,"CC0-1.0",57.79,40.64,55.69,38.59 -ViTSmall16_Weights.LANDSAT_ETM_SR_MOCO,7,6,`link `__,`link `__,"CC0-1.0",64.09,47.21,52.37,35.48 -ViTSmall16_Weights.LANDSAT_ETM_SR_SIMCLR,7,6,`link `__,`link `__,"CC0-1.0",63.99,47.05,53.17,36.21 -ResNet18_Weights.LANDSAT_OLI_TIRS_TOA_MOCO,8--9,11,`link `__,`link `__,"CC0-1.0",67.82,51.30,65.74,48.96 -ResNet18_Weights.LANDSAT_OLI_TIRS_TOA_SIMCLR,8--9,11,`link `__,`link `__,"CC0-1.0",62.14,45.08,60.01,42.86 -ResNet50_Weights.LANDSAT_OLI_TIRS_TOA_MOCO,8--9,11,`link `__,`link `__,"CC0-1.0",69.17,52.87,67.29,50.70 -ResNet50_Weights.LANDSAT_OLI_TIRS_TOA_SIMCLR,8--9,11,`link `__,`link `__,"CC0-1.0",64.66,47.78,62.08,45.01 -ViTSmall16_Weights.LANDSAT_OLI_TIRS_TOA_MOCO,8--9,11,`link `__,`link `__,"CC0-1.0",67.11,50.49,64.62,47.73 -ViTSmall16_Weights.LANDSAT_OLI_TIRS_TOA_SIMCLR,8--9,11,`link `__,`link `__,"CC0-1.0",66.12,49.39,63.88,46.94 -ResNet18_Weights.LANDSAT_OLI_SR_MOCO,8--9,7,`link `__,`link `__,"CC0-1.0",67.01,50.39,68.05,51.57 -ResNet18_Weights.LANDSAT_OLI_SR_SIMCLR,8--9,7,`link `__,`link `__,"CC0-1.0",59.93,42.79,57.44,40.30 -ResNet50_Weights.LANDSAT_OLI_SR_MOCO,8--9,7,`link `__,`link `__,"CC0-1.0",67.44,50.88,65.96,49.21 -ResNet50_Weights.LANDSAT_OLI_SR_SIMCLR,8--9,7,`link `__,`link `__,"CC0-1.0",63.65,46.68,60.01,43.17 -ViTSmall16_Weights.LANDSAT_OLI_SR_MOCO,8--9,7,`link `__,`link `__,"CC0-1.0",66.81,50.16,64.17,47.24 -ViTSmall16_Weights.LANDSAT_OLI_SR_SIMCLR,8--9,7,`link `__,`link `__,"CC0-1.0",65.04,48.20,62.61,45.46 +ResNet18_Weights.LANDSAT_TM_TOA_MOCO,4--5,5,`link `__,`link `__,"CC0-1.0",67.65,51.11,68.70,52.32 +ResNet18_Weights.LANDSAT_TM_TOA_SIMCLR,4--5,5,`link `__,`link `__,"CC0-1.0",60.86,43.74,61.94,44.86 +ResNet50_Weights.LANDSAT_TM_TOA_MOCO,4--5,5,`link `__,`link `__,"CC0-1.0",68.75,53.28,69.45,53.20 +ResNet50_Weights.LANDSAT_TM_TOA_SIMCLR,4--5,5,`link `__,`link `__,"CC0-1.0",62.05,44.98,62.80,45.77 +ViTSmall16_Weights.LANDSAT_TM_TOA_MOCO,4--5,5,`link `__,`link `__,"CC0-1.0",67.17,50.57,67.60,51.07 +ViTSmall16_Weights.LANDSAT_TM_TOA_SIMCLR,4--5,5,`link `__,`link `__,"CC0-1.0",66.82,50.17,66.92,50.28 +ResNet18_Weights.LANDSAT_ETM_TOA_MOCO,7,9,`link `__,`link `__,"CC0-1.0",65.22,48.39,62.84,45.81 +ResNet18_Weights.LANDSAT_ETM_TOA_SIMCLR,7,9,`link `__,`link `__,"CC0-1.0",58.76,41.60,56.47,39.34 +ResNet50_Weights.LANDSAT_ETM_TOA_MOCO,7,9,`link `__,`link `__,"CC0-1.0",66.60,49.92,64.12,47.19 +ResNet50_Weights.LANDSAT_ETM_TOA_SIMCLR,7,9,`link `__,`link `__,"CC0-1.0",57.17,40.02,54.95,37.88 +ViTSmall16_Weights.LANDSAT_ETM_TOA_MOCO,7,9,`link `__,`link `__,"CC0-1.0",63.75,46.79,60.88,43.70 +ViTSmall16_Weights.LANDSAT_ETM_TOA_SIMCLR,7,9,`link `__,`link `__,"CC0-1.0",63.33,46.34,59.06,41.91 +ResNet18_Weights.LANDSAT_ETM_SR_MOCO,7,6,`link `__,`link `__,"CC0-1.0",64.18,47.25,67.30,50.71 +ResNet18_Weights.LANDSAT_ETM_SR_SIMCLR,7,6,`link `__,`link `__,"CC0-1.0",57.26,40.11,54.42,37.48 +ResNet50_Weights.LANDSAT_ETM_SR_MOCO,7,6,`link `__,`link `__,"CC0-1.0",64.37,47.46,62.35,45.30 +ResNet50_Weights.LANDSAT_ETM_SR_SIMCLR,7,6,`link `__,`link `__,"CC0-1.0",57.79,40.64,55.69,38.59 +ViTSmall16_Weights.LANDSAT_ETM_SR_MOCO,7,6,`link `__,`link `__,"CC0-1.0",64.09,47.21,52.37,35.48 +ViTSmall16_Weights.LANDSAT_ETM_SR_SIMCLR,7,6,`link `__,`link `__,"CC0-1.0",63.99,47.05,53.17,36.21 +ResNet18_Weights.LANDSAT_OLI_TIRS_TOA_MOCO,8--9,11,`link `__,`link `__,"CC0-1.0",67.82,51.30,65.74,48.96 +ResNet18_Weights.LANDSAT_OLI_TIRS_TOA_SIMCLR,8--9,11,`link `__,`link `__,"CC0-1.0",62.14,45.08,60.01,42.86 +ResNet50_Weights.LANDSAT_OLI_TIRS_TOA_MOCO,8--9,11,`link `__,`link `__,"CC0-1.0",69.17,52.87,67.29,50.70 +ResNet50_Weights.LANDSAT_OLI_TIRS_TOA_SIMCLR,8--9,11,`link `__,`link `__,"CC0-1.0",64.66,47.78,62.08,45.01 +ViTSmall16_Weights.LANDSAT_OLI_TIRS_TOA_MOCO,8--9,11,`link `__,`link `__,"CC0-1.0",67.11,50.49,64.62,47.73 +ViTSmall16_Weights.LANDSAT_OLI_TIRS_TOA_SIMCLR,8--9,11,`link `__,`link `__,"CC0-1.0",66.12,49.39,63.88,46.94 +ResNet18_Weights.LANDSAT_OLI_SR_MOCO,8--9,7,`link `__,`link `__,"CC0-1.0",67.01,50.39,68.05,51.57 +ResNet18_Weights.LANDSAT_OLI_SR_SIMCLR,8--9,7,`link `__,`link `__,"CC0-1.0",59.93,42.79,57.44,40.30 +ResNet50_Weights.LANDSAT_OLI_SR_MOCO,8--9,7,`link `__,`link `__,"CC0-1.0",67.44,50.88,65.96,49.21 +ResNet50_Weights.LANDSAT_OLI_SR_SIMCLR,8--9,7,`link `__,`link `__,"CC0-1.0",63.65,46.68,60.01,43.17 +ViTSmall16_Weights.LANDSAT_OLI_SR_MOCO,8--9,7,`link `__,`link `__,"CC0-1.0",66.81,50.16,64.17,47.24 +ViTSmall16_Weights.LANDSAT_OLI_SR_SIMCLR,8--9,7,`link `__,`link `__,"CC0-1.0",65.04,48.20,62.61,45.46 Swin_V2_B_Weights.LANDSAT_MS_SI_SATLAS,11,'link `__,`link `__,"ODC-BY",,,, diff --git a/torchgeo/datasets/ssl4eo.py b/torchgeo/datasets/ssl4eo.py index e8779a3e5f5..7b862030e63 100644 --- a/torchgeo/datasets/ssl4eo.py +++ b/torchgeo/datasets/ssl4eo.py @@ -88,7 +88,7 @@ class SSL4EOL(NonGeoDataset): If you use this dataset in your research, please cite the following paper: - * https://arxiv.org/abs/2306.09424 + * https://proceedings.neurips.cc/paper_files/paper/2023/hash/bbf7ee04e2aefec136ecf60e346c2e61-Abstract-Datasets_and_Benchmarks.html .. versionadded:: 0.5 """ # noqa: E501 diff --git a/torchgeo/datasets/ssl4eo_benchmark.py b/torchgeo/datasets/ssl4eo_benchmark.py index 8f2858b7e23..d7d2444eb10 100644 --- a/torchgeo/datasets/ssl4eo_benchmark.py +++ b/torchgeo/datasets/ssl4eo_benchmark.py @@ -42,10 +42,10 @@ class SSL4EOLBenchmark(NonGeoDataset): If you use this dataset in your research, please cite the following paper: - * https://arxiv.org/abs/2306.09424 + * https://proceedings.neurips.cc/paper_files/paper/2023/hash/bbf7ee04e2aefec136ecf60e346c2e61-Abstract-Datasets_and_Benchmarks.html .. versionadded:: 0.5 - """ + """ # noqa: E501 url = "https://hf.co/datasets/torchgeo/ssl4eo-l-benchmark/resolve/da96ae2b04cb509710b72fce9131c2a3d5c211c2/{}.tar.gz" # noqa: E501