зеркало из https://github.com/microsoft/torchgeo.git
SSL4EO-L Weights (#1482)
* ssl4eo landsat 18 weights * add resnet and vit moco weights under new hf repo * add simclr ssl checkpoint * add docs * order models and new csv tablesf for docs * adjust spacing of table * lots of numbers and formatting * missing resnet etm-sr weights * fix failing tests * check slow tests * Fix URL checksum * fix weight names * Use 2 digits for all metrics * hyphen -> en dash * Use 2 digits for all metrics * Landsat first * EOL char before EOF char --------- Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
This commit is contained in:
Nils Lehmann
GitHub
Родитель
f8f3f8a541
Коммит
18139b440b
|
|
@ -0,0 +1,31 @@
|
|||
Weight,Landsat,Channels,Source,Citation,NLCD (Acc),NLCD (mIoU),CDL (Acc),CDL (mIoU)
|
||||
ResNet18_Weights.LANDSAT_TM_TOA_MOCO,4--5,5,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,67.65,51.11,68.70,52.32
|
||||
ResNet18_Weights.LANDSAT_TM_TOA_SIMCLR,4--5,5,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,60.86,43.74,61.94,44.86
|
||||
ResNet50_Weights.LANDSAT_TM_TOA_MOCO,4--5,5,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,68.75,53.28,69.45,53.20
|
||||
ResNet50_Weights.LANDSAT_TM_TOA_SIMCLR,4--5,5,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,62.05,44.98,62.80,45.77
|
||||
ViTSmall16_Weights.LANDSAT_TM_TOA_MOCO,4--5,5,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,67.17,50.57,67.60,51.07
|
||||
ViTSmall16_Weights.LANDSAT_TM_TOA_SIMCLR,4--5,5,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,66.82,50.17,66.92,50.28
|
||||
ResNet18_Weights.LANDSAT_ETM_TOA_MOCO,7,9,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,65.22,48.39,62.84,45.81
|
||||
ResNet18_Weights.LANDSAT_ETM_TOA_SIMCLR,7,9,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,58.76,41.60,56.47,39.34
|
||||
ResNet50_Weights.LANDSAT_ETM_TOA_MOCO,7,9,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,66.60,49.92,64.12,47.19
|
||||
ResNet50_Weights.LANDSAT_ETM_TOA_SIMCLR,7,9,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,57.17,40.02,54.95,37.88
|
||||
ViTSmall16_Weights.LANDSAT_ETM_TOA_MOCO,7,9,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,63.75,46.79,60.88,43.70
|
||||
ViTSmall16_Weights.LANDSAT_ETM_TOA_SIMCLR,7,9,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,63.33,46.34,59.06,41.91
|
||||
ResNet18_Weights.LANDSAT_ETM_SR_MOCO,7,6,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,64.18,47.25,67.30,50.71
|
||||
ResNet18_Weights.LANDSAT_ETM_SR_SIMCLR,7,6,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,57.26,40.11,54.42,37.48
|
||||
ResNet50_Weights.LANDSAT_ETM_SR_MOCO,7,6,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,64.37,47.46,62.35,45.30
|
||||
ResNet50_Weights.LANDSAT_ETM_SR_SIMCLR,7,6,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,57.79,40.64,55.69,38.59
|
||||
ViTSmall16_Weights.LANDSAT_ETM_SR_MOCO,7,6,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,64.09,47.21,52.37,35.48
|
||||
ViTSmall16_Weights.LANDSAT_ETM_SR_SIMCLR,7,6,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,63.99,47.05,53.17,36.21
|
||||
ResNet18_Weights.LANDSAT_OLI_TIRS_TOA_MOCO,8--9,11,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,67.82,51.30,65.74,48.96
|
||||
ResNet18_Weights.LANDSAT_OLI_TIRS_TOA_SIMCLR,8--9,11,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,62.14,45.08,60.01,42.86
|
||||
ResNet50_Weights.LANDSAT_OLI_TIRS_TOA_MOCO,8--9,11,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,69.17,52.87,67.29,50.70
|
||||
ResNet50_Weights.LANDSAT_OLI_TIRS_TOA_SIMCLR,8--9,11,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,64.66,47.78,62.08,45.01
|
||||
ViTSmall16_Weights.LANDSAT_OLI_TIRS_TOA_MOCO,8--9,11,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,67.11,50.49,64.62,47.73
|
||||
ViTSmall16_Weights.LANDSAT_OLI_TIRS_TOA_SIMCLR,8--9,11,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,66.12,49.39,63.88,46.94
|
||||
ResNet18_Weights.LANDSAT_OLI_SR_MOCO,8--9,7,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,67.01,50.39,68.05,51.57
|
||||
ResNet18_Weights.LANDSAT_OLI_SR_SIMCLR,8--9,7,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,59.93,42.79,57.44,40.30
|
||||
ResNet50_Weights.LANDSAT_OLI_SR_MOCO,8--9,7,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,67.44,50.88,65.96,49.21
|
||||
ResNet50_Weights.LANDSAT_OLI_SR_SIMCLR,8--9,7,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,63.65,46.68,60.01,43.17
|
||||
ViTSmall16_Weights.LANDSAT_OLI_SR_MOCO,8--9,7,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,66.81,50.16,64.17,47.24
|
||||
ViTSmall16_Weights.LANDSAT_OLI_SR_SIMCLR,8--9,7,`link <https://github.com/microsoft/torchgeo>`__,`link <https://arxiv.org/abs/2306.09424>`__,65.04,48.20,62.61,45.46
|
||||
|
|
|
@ -0,0 +1,2 @@
|
|||
Weight,Channels,Source,Citation
|
||||
ResNet50_Weights.FMOW_RGB_GASSL, 3,`link <https://github.com/sustainlab-group/geography-aware-ssl>`__,`link <https://arxiv.org/abs/2011.09980>`__
|
||||
|
|
|
@ -39,24 +39,12 @@ ResNet
|
|||
.. autoclass:: ResNet18_Weights
|
||||
.. autoclass:: ResNet50_Weights
|
||||
|
||||
.. csv-table::
|
||||
:widths: 45 10 10 10 15 10 10 10
|
||||
:header-rows: 1
|
||||
:align: center
|
||||
:file: resnet_pretrained_weights.csv
|
||||
|
||||
Vision Transformer
|
||||
^^^^^^^^^^^^^^^^^^
|
||||
|
||||
.. autofunction:: vit_small_patch16_224
|
||||
.. autoclass:: ViTSmall16_Weights
|
||||
|
||||
.. csv-table::
|
||||
:widths: 45 10 10 10 15 10 10 10
|
||||
:header-rows: 1
|
||||
:align: center
|
||||
:file: vit_pretrained_weights.csv
|
||||
|
||||
Utility Functions
|
||||
^^^^^^^^^^^^^^^^^
|
||||
|
||||
|
|
@ -64,3 +52,45 @@ Utility Functions
|
|||
.. autofunction:: get_model_weights
|
||||
.. autofunction:: get_weight
|
||||
.. autofunction:: list_models
|
||||
|
||||
|
||||
Pretrained Weights
|
||||
^^^^^^^^^^^^^^^^^^
|
||||
|
||||
Landsat
|
||||
-------
|
||||
|
||||
.. csv-table::
|
||||
:widths: 65 10 10 10 10 10 10 10 10
|
||||
:header-rows: 1
|
||||
:align: center
|
||||
:file: landsat_pretrained_weights.csv
|
||||
|
||||
|
||||
Sentinel-1
|
||||
----------
|
||||
|
||||
.. csv-table::
|
||||
:widths: 45 10 10 10
|
||||
:header-rows: 1
|
||||
:align: center
|
||||
:file: sentinel1_pretrained_weights.csv
|
||||
|
||||
|
||||
Sentinel-2
|
||||
----------
|
||||
|
||||
.. csv-table::
|
||||
:widths: 45 10 10 10 15 10 10 10
|
||||
:header-rows: 1
|
||||
:align: center
|
||||
:file: sentinel2_pretrained_weights.csv
|
||||
|
||||
Other Data Sources
|
||||
------------------
|
||||
|
||||
.. csv-table::
|
||||
:widths: 45 10 10 10
|
||||
:header-rows: 1
|
||||
:align: center
|
||||
:file: misc_pretrained_weights.csv
|
||||
|
|
|
|||
|
|
@ -0,0 +1,2 @@
|
|||
Weight,Channels,Source,Citation
|
||||
ResNet50_Weights.SENTINEL1_ALL_MOCO, 2,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__
|
||||
|
|
|
@ -2,9 +2,9 @@ Weight,Channels,Source,Citation,BigEarthNet,EuroSAT,So2Sat,OSCD
|
|||
ResNet18_Weights.SENTINEL2_ALL_MOCO,13,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,,,,
|
||||
ResNet18_Weights.SENTINEL2_RGB_MOCO, 3,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,,,,
|
||||
ResNet18_Weights.SENTINEL2_RGB_SECO, 3,`link <https://github.com/ServiceNow/seasonal-contrast>`__,`link <https://arxiv.org/abs/2103.16607>`__,87.27,93.14,,46.94
|
||||
ResNet50_Weights.FMOW_RGB_GASSL, 3,`link <https://github.com/sustainlab-group/geography-aware-ssl>`__,`link <https://arxiv.org/abs/2011.09980>`__,,,,
|
||||
ResNet50_Weights.SENTINEL1_ALL_MOCO, 2,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,,,,
|
||||
ResNet50_Weights.SENTINEL2_ALL_DINO,13,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,90.7,99.1,63.6,
|
||||
ResNet50_Weights.SENTINEL2_ALL_MOCO,13,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,91.8,99.1,60.9,
|
||||
ResNet50_Weights.SENTINEL2_RGB_MOCO, 3,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,,,,
|
||||
ResNet50_Weights.SENTINEL2_RGB_SECO, 3,`link <https://github.com/ServiceNow/seasonal-contrast>`__,`link <https://arxiv.org/abs/2103.16607>`__,87.81,,,
|
||||
ViTSmall16_Weights.SENTINEL2_ALL_DINO,13,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,90.5,99.0,62.2,
|
||||
ViTSmall16_Weights.SENTINEL2_ALL_MOCO,13,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,89.9,98.6,61.6,
|
||||
|
|
|
@ -1,3 +0,0 @@
|
|||
Weight,Channels,Source,Citation,BigEarthNet,EuroSAT,So2Sat,OSCD
|
||||
ViTSmall16_Weights.SENTINEL2_ALL_DINO,13,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,90.5,99.0,62.2,
|
||||
ViTSmall16_Weights.SENTINEL2_ALL_MOCO,13,`link <https://github.com/zhu-xlab/SSL4EO-S12>`__,`link <https://arxiv.org/abs/2211.07044>`__,89.9,98.6,61.6,
|
||||
|
|
|
@ -47,7 +47,9 @@ class TestResNet18:
|
|||
|
||||
def test_transforms(self, mocked_weights: WeightsEnum) -> None:
|
||||
c = mocked_weights.meta["in_chans"]
|
||||
sample = {"image": torch.arange(c * 4 * 4, dtype=torch.float).view(c, 4, 4)}
|
||||
sample = {
|
||||
"image": torch.arange(c * 224 * 224, dtype=torch.float).view(c, 224, 224)
|
||||
}
|
||||
mocked_weights.transforms(sample)
|
||||
|
||||
@pytest.mark.slow
|
||||
|
|
@ -82,7 +84,9 @@ class TestResNet50:
|
|||
|
||||
def test_transforms(self, mocked_weights: WeightsEnum) -> None:
|
||||
c = mocked_weights.meta["in_chans"]
|
||||
sample = {"image": torch.arange(c * 4 * 4, dtype=torch.float).view(c, 4, 4)}
|
||||
sample = {
|
||||
"image": torch.arange(c * 224 * 224, dtype=torch.float).view(c, 224, 224)
|
||||
}
|
||||
mocked_weights.transforms(sample)
|
||||
|
||||
@pytest.mark.slow
|
||||
|
|
|
|||
|
|
@ -49,7 +49,9 @@ class TestViTSmall16:
|
|||
|
||||
def test_transforms(self, mocked_weights: WeightsEnum) -> None:
|
||||
c = mocked_weights.meta["in_chans"]
|
||||
sample = {"image": torch.arange(c * 4 * 4, dtype=torch.float).view(c, 4, 4)}
|
||||
sample = {
|
||||
"image": torch.arange(c * 224 * 224, dtype=torch.float).view(c, 224, 224)
|
||||
}
|
||||
mocked_weights.transforms(sample)
|
||||
|
||||
@pytest.mark.slow
|
||||
|
|
|
|||
|
|
@ -52,6 +52,13 @@ _gassl_transforms = AugmentationSequential(
|
|||
data_keys=["image"],
|
||||
)
|
||||
|
||||
# https://github.com/microsoft/torchgeo/blob/8b53304d42c269f9001cb4e861a126dc4b462606/torchgeo/datamodules/ssl4eo_benchmark.py#L43 # noqa: E501
|
||||
_ssl4eo_l_transforms = AugmentationSequential(
|
||||
K.Normalize(mean=torch.tensor(0), std=torch.tensor(255)),
|
||||
K.CenterCrop((224, 224)),
|
||||
data_keys=["image"],
|
||||
)
|
||||
|
||||
# https://github.com/pytorch/vision/pull/6883
|
||||
# https://github.com/pytorch/vision/pull/7107
|
||||
# Can be removed once torchvision>=0.15 is required
|
||||
|
|
@ -67,6 +74,136 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
|
|||
.. versionadded:: 0.4
|
||||
"""
|
||||
|
||||
LANDSAT_TM_TOA_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_tm_toa_moco-1c691b4f.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_TM_TOA_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_tm_toa_simclr-d2d38ace.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_TOA_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_etm_toa_moco-bb88689c.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 9,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_TOA_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_etm_toa_simclr-4d813f79.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 9,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_SR_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_etm_sr_moco-4f078acd.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 6,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_SR_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_etm_sr_simclr-8e8543b4.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 6,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_TIRS_TOA_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_oli_tirs_toa_moco-a3002f51.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 11,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_TIRS_TOA_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_oli_tirs_toa_simclr-b0635cc6.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 11,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_SR_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_oli_sr_moco-660e82ed.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_SR_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_oli_sr_simclr-7bced5be.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
SENTINEL2_ALL_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/resnet18_sentinel2_all_moco/resolve/main/resnet18_sentinel2_all_moco-59bfdff9.pth", # noqa: E501
|
||||
transforms=_zhu_xlab_transforms,
|
||||
|
|
@ -129,6 +266,136 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
|
|||
},
|
||||
)
|
||||
|
||||
LANDSAT_TM_TOA_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_tm_toa_moco-ba1ce753.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "resnet50",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_TM_TOA_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_tm_toa_simclr-a1c93432.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "resnet50",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_TOA_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_etm_toa_moco-e9a84d5a.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 9,
|
||||
"model": "resnet50",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_TOA_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_etm_toa_simclr-70b5575f.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 9,
|
||||
"model": "resnet50",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_SR_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_etm_sr_moco-1266cde3.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 6,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_SR_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_etm_sr_simclr-e5d185d7.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 6,
|
||||
"model": "resnet18",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_TIRS_TOA_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_oli_tirs_toa_moco-de7f5e0f.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 11,
|
||||
"model": "resnet50",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_TIRS_TOA_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_oli_tirs_toa_simclr-030cebfe.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 11,
|
||||
"model": "resnet50",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_SR_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_oli_sr_moco-ff580dad.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "resnet50",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_SR_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_oli_sr_simclr-94f78913.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "resnet50",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
SENTINEL1_ALL_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/resnet50_sentinel1_all_moco/resolve/main/resnet50_sentinel1_all_moco-906e4356.pth", # noqa: E501
|
||||
transforms=_zhu_xlab_transforms,
|
||||
|
|
|
|||
|
|
@ -25,6 +25,13 @@ _zhu_xlab_transforms = AugmentationSequential(
|
|||
data_keys=["image"],
|
||||
)
|
||||
|
||||
# https://github.com/microsoft/torchgeo/blob/8b53304d42c269f9001cb4e861a126dc4b462606/torchgeo/datamodules/ssl4eo_benchmark.py#L43 # noqa: E501
|
||||
_ssl4eo_l_transforms = AugmentationSequential(
|
||||
K.Normalize(mean=torch.tensor(0), std=torch.tensor(255)),
|
||||
K.CenterCrop((224, 224)),
|
||||
data_keys=["image"],
|
||||
)
|
||||
|
||||
# https://github.com/pytorch/vision/pull/6883
|
||||
# https://github.com/pytorch/vision/pull/7107
|
||||
# Can be removed once torchvision>=0.15 is required
|
||||
|
|
@ -40,6 +47,136 @@ class ViTSmall16_Weights(WeightsEnum): # type: ignore[misc]
|
|||
.. versionadded:: 0.4
|
||||
"""
|
||||
|
||||
LANDSAT_TM_TOA_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_tm_toa_moco-a1c967d8.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_TM_TOA_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_tm_toa_simclr-7c2d9799.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_TOA_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_etm_toa_moco-26d19bcf.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 9,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_TOA_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_etm_toa_simclr-34fb12cb.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 9,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_SR_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_etm_sr_moco-eaa4674e.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 6,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_ETM_SR_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_etm_sr_simclr-a14c466a.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 6,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_TIRS_TOA_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_oli_tirs_toa_moco-c7c2cceb.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 11,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_TIRS_TOA_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_oli_tirs_toa_simclr-ad43e9a4.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 11,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_SR_MOCO = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_oli_sr_moco-c9b8898d.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "moco",
|
||||
},
|
||||
)
|
||||
|
||||
LANDSAT_OLI_SR_SIMCLR = Weights(
|
||||
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/vits16_landsat_oli_sr_simclr-4e8f6102.pth", # noqa: E501
|
||||
transforms=_ssl4eo_l_transforms,
|
||||
meta={
|
||||
"dataset": "SSL4EO-L",
|
||||
"in_chans": 7,
|
||||
"model": "vit_small_patch16_224",
|
||||
"publication": "https://arxiv.org/abs/2306.09424",
|
||||
"repo": "https://github.com/microsoft/torchgeo",
|
||||
"ssl_method": "simclr",
|
||||
},
|
||||
)
|
||||
|
||||
SENTINEL2_ALL_DINO = Weights(
|
||||
url="https://huggingface.co/torchgeo/vit_small_patch16_224_sentinel2_all_dino/resolve/main/vit_small_patch16_224_sentinel2_all_dino-36bcc127.pth", # noqa: E501
|
||||
transforms=_zhu_xlab_transforms,
|
||||
|
|
|
|||
Загрузка…
Ссылка в новой задаче