Skip to content

Commit

Permalink
HuggingFace: use stable download URLs (#1916)
Browse files Browse the repository at this point in the history
  • Loading branch information
adamjstewart authored and isaaccorley committed Mar 2, 2024
1 parent 46f5fe4 commit d64d9b1
Show file tree
Hide file tree
Showing 11 changed files with 54 additions and 54 deletions.
10 changes: 5 additions & 5 deletions torchgeo/datasets/eurosat.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ class EuroSAT(NonGeoClassificationDataset):
* https://ieeexplore.ieee.org/document/8519248
"""

url = "https://huggingface.co/datasets/torchgeo/eurosat/resolve/main/EuroSATallBands.zip" # noqa: E501
url = "https://hf.co/datasets/torchgeo/eurosat/resolve/06fd1b090bceecc0ce724cd21578ba7a6664fe8d/EuroSATallBands.zip" # noqa: E501
filename = "EuroSATallBands.zip"
md5 = "5ac12b3b2557aa56e1826e981e8e200e"

Expand Down Expand Up @@ -310,14 +310,14 @@ class EuroSAT100(EuroSAT):
.. versionadded:: 0.5
"""

url = "https://huggingface.co/datasets/torchgeo/eurosat/resolve/main/EuroSAT100.zip"
url = "https://hf.co/datasets/torchgeo/eurosat/resolve/06fd1b090bceecc0ce724cd21578ba7a6664fe8d/EuroSAT100.zip" # noqa: E501
filename = "EuroSAT100.zip"
md5 = "c21c649ba747e86eda813407ef17d596"

split_urls = {
"train": "https://huggingface.co/datasets/torchgeo/eurosat/raw/main/eurosat-train.txt", # noqa: E501
"val": "https://huggingface.co/datasets/torchgeo/eurosat/raw/main/eurosat-val.txt", # noqa: E501
"test": "https://huggingface.co/datasets/torchgeo/eurosat/raw/main/eurosat-test.txt", # noqa: E501
"train": "https://hf.co/datasets/torchgeo/eurosat/raw/06fd1b090bceecc0ce724cd21578ba7a6664fe8d/eurosat-train.txt", # noqa: E501
"val": "https://hf.co/datasets/torchgeo/eurosat/raw/06fd1b090bceecc0ce724cd21578ba7a6664fe8d/eurosat-val.txt", # noqa: E501
"test": "https://hf.co/datasets/torchgeo/eurosat/raw/06fd1b090bceecc0ce724cd21578ba7a6664fe8d/eurosat-test.txt", # noqa: E501
}
split_md5s = {
"train": "033d0c23e3a75e3fa79618b0e35fe1c7",
Expand Down
2 changes: 1 addition & 1 deletion torchgeo/datasets/l7irish.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ class L7Irish(RasterDataset):
.. versionadded:: 0.5
""" # noqa: E501

url = "https://huggingface.co/datasets/torchgeo/l7irish/resolve/main/{}.tar.gz" # noqa: E501
url = "https://hf.co/datasets/torchgeo/l7irish/resolve/6807e0b22eca7f9a8a3903ea673b31a115837464/{}.tar.gz" # noqa: E501

md5s = {
"austral": "0a34770b992a62abeb88819feb192436",
Expand Down
2 changes: 1 addition & 1 deletion torchgeo/datasets/l8biome.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ class L8Biome(RasterDataset):
.. versionadded:: 0.5
""" # noqa: E501

url = "https://huggingface.co/datasets/torchgeo/l8biome/resolve/main/{}.tar.gz" # noqa: E501
url = "https://hf.co/datasets/torchgeo/l8biome/resolve/f76df19accce34d2acc1878d88b9491bc81f94c8/{}.tar.gz" # noqa: E501

md5s = {
"barren": "0eb691822d03dabd4f5ea8aadd0b41c3",
Expand Down
2 changes: 1 addition & 1 deletion torchgeo/datasets/mapinwild.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ class MapInWild(NonGeoDataset):
.. versionadded:: 0.5
"""

url = "https://huggingface.co/datasets/burakekim/mapinwild/resolve/main/"
url = "https://hf.co/datasets/burakekim/mapinwild/resolve/d963778e31e7e0ed2329c0f4cbe493be532f0e71/" # noqa: E501

modality_urls = {
"esa_wc": {"esa_wc/ESA_WC.zip"},
Expand Down
2 changes: 1 addition & 1 deletion torchgeo/datasets/skippd.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ class SKIPPD(NonGeoDataset):
.. versionadded:: 0.5
"""

url = "https://huggingface.co/datasets/torchgeo/skippd/resolve/main/{}"
url = "https://hf.co/datasets/torchgeo/skippd/resolve/a16c7e200b4618cd93be3143cdb973e3f21498fa/{}" # noqa: E501
md5 = {
"forecast": "f4f3509ddcc83a55c433be9db2e51077",
"nowcast": "0000761d403e45bb5f86c21d3c69aa80",
Expand Down
2 changes: 1 addition & 1 deletion torchgeo/datasets/ssl4eo.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ class _Metadata(TypedDict):
"oli_sr": {"num_bands": 7, "rgb_bands": [3, 2, 1]},
}

url = "https://hf.co/datasets/torchgeo/ssl4eo_l/resolve/main/{0}/ssl4eo_l_{0}.tar.gz{1}" # noqa: E501
url = "https://hf.co/datasets/torchgeo/ssl4eo_l/resolve/e2467887e6a6bcd7547d9d5999f8d9bc3323dc31/{0}/ssl4eo_l_{0}.tar.gz{1}" # noqa: E501
checksums = {
"tm_toa": {
"aa": "553795b8d73aa253445b1e67c5b81f11",
Expand Down
2 changes: 1 addition & 1 deletion torchgeo/datasets/ssl4eo_benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ class SSL4EOLBenchmark(NonGeoDataset):
.. versionadded:: 0.5
"""

url = "https://huggingface.co/datasets/torchgeo/ssl4eo-l-benchmark/resolve/main/{}.tar.gz" # noqa: E501
url = "https://hf.co/datasets/torchgeo/ssl4eo-l-benchmark/resolve/da96ae2b04cb509710b72fce9131c2a3d5c211c2/{}.tar.gz" # noqa: E501

valid_sensors = ["tm_toa", "etm_toa", "etm_sr", "oli_tirs_toa", "oli_sr"]
valid_products = ["cdl", "nlcd"]
Expand Down
2 changes: 1 addition & 1 deletion torchgeo/datasets/ucmerced.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ class UCMerced(NonGeoClassificationDataset):
* https://dl.acm.org/doi/10.1145/1869790.1869829
"""

url = "https://huggingface.co/datasets/torchgeo/ucmerced/resolve/main/UCMerced_LandUse.zip" # noqa: E501
url = "https://hf.co/datasets/torchgeo/ucmerced/resolve/d0af6e2eeea2322af86078068bd83337148a2149/UCMerced_LandUse.zip" # noqa: E501
filename = "UCMerced_LandUse.zip"
md5 = "5b7ec56793786b6dc8a908e8854ac0e4"

Expand Down
2 changes: 1 addition & 1 deletion torchgeo/datasets/usavars.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ class USAVars(NonGeoDataset):
.. versionadded:: 0.3
"""

data_url = "https://huggingface.co/datasets/torchgeo/usavars/resolve/main/{}"
data_url = "https://hf.co/datasets/torchgeo/usavars/resolve/01377abfaf50c0cc8548aaafb79533666bbf288f/{}" # noqa: E501
dirname = "uar"

md5 = "677e89fd20e5dd0fe4d29b61827c2456"
Expand Down
58 changes: 29 additions & 29 deletions torchgeo/models/resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
"""

LANDSAT_TM_TOA_MOCO = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_tm_toa_moco-1c691b4f.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_tm_toa_moco-1c691b4f.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -88,7 +88,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_TM_TOA_SIMCLR = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_tm_toa_simclr-d2d38ace.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_tm_toa_simclr-d2d38ace.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -101,7 +101,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_ETM_TOA_MOCO = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_etm_toa_moco-bb88689c.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_etm_toa_moco-bb88689c.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -114,7 +114,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_ETM_TOA_SIMCLR = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_etm_toa_simclr-4d813f79.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_etm_toa_simclr-4d813f79.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -127,7 +127,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_ETM_SR_MOCO = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_etm_sr_moco-4f078acd.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_etm_sr_moco-4f078acd.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -140,7 +140,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_ETM_SR_SIMCLR = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_etm_sr_simclr-8e8543b4.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_etm_sr_simclr-8e8543b4.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -153,7 +153,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

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
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_oli_tirs_toa_moco-a3002f51.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -166,7 +166,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

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
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_oli_tirs_toa_simclr-b0635cc6.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -179,7 +179,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_OLI_SR_MOCO = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_oli_sr_moco-660e82ed.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_oli_sr_moco-660e82ed.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -192,7 +192,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_OLI_SR_SIMCLR = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet18_landsat_oli_sr_simclr-7bced5be.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet18_landsat_oli_sr_simclr-7bced5be.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -205,7 +205,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

SENTINEL2_ALL_MOCO = Weights(
url="https://huggingface.co/torchgeo/resnet18_sentinel2_all_moco/resolve/main/resnet18_sentinel2_all_moco-59bfdff9.pth", # noqa: E501
url="https://hf.co/torchgeo/resnet18_sentinel2_all_moco/resolve/5b8cddc9a14f3844350b7f40b85bcd32aed75918/resnet18_sentinel2_all_moco-59bfdff9.pth", # noqa: E501
transforms=_zhu_xlab_transforms,
meta={
"dataset": "SSL4EO-S12",
Expand All @@ -218,7 +218,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

SENTINEL2_RGB_MOCO = Weights(
url="https://huggingface.co/torchgeo/resnet18_sentinel2_rgb_moco/resolve/main/resnet18_sentinel2_rgb_moco-e3a335e3.pth", # noqa: E501
url="https://hf.co/torchgeo/resnet18_sentinel2_rgb_moco/resolve/e1c032e7785fd0625224cdb6699aa138bb304eec/resnet18_sentinel2_rgb_moco-e3a335e3.pth", # noqa: E501
transforms=_zhu_xlab_transforms,
meta={
"dataset": "SSL4EO-S12",
Expand All @@ -231,7 +231,7 @@ class ResNet18_Weights(WeightsEnum): # type: ignore[misc]
)

SENTINEL2_RGB_SECO = Weights(
url="https://huggingface.co/torchgeo/resnet18_sentinel2_rgb_seco/resolve/main/resnet18_sentinel2_rgb_seco-cefca942.pth", # noqa: E501
url="https://hf.co/torchgeo/resnet18_sentinel2_rgb_seco/resolve/f8dcee692cf7142163b55a5c197d981fe0e717a0/resnet18_sentinel2_rgb_seco-cefca942.pth", # noqa: E501
transforms=_seco_transforms,
meta={
"dataset": "SeCo Dataset",
Expand All @@ -254,7 +254,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
"""

FMOW_RGB_GASSL = Weights(
url="https://huggingface.co/torchgeo/resnet50_fmow_rgb_gassl/resolve/main/resnet50_fmow_rgb_gassl-da43d987.pth", # noqa: E501
url="https://hf.co/torchgeo/resnet50_fmow_rgb_gassl/resolve/fe8a91026cf9104f1e884316b8e8772d7af9052c/resnet50_fmow_rgb_gassl-da43d987.pth", # noqa: E501
transforms=_gassl_transforms,
meta={
"dataset": "fMoW Dataset",
Expand All @@ -267,7 +267,7 @@ 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
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_tm_toa_moco-ba1ce753.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -280,7 +280,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_TM_TOA_SIMCLR = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_tm_toa_simclr-a1c93432.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_tm_toa_simclr-a1c93432.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -293,7 +293,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_ETM_TOA_MOCO = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_etm_toa_moco-e9a84d5a.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_etm_toa_moco-e9a84d5a.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -306,7 +306,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_ETM_TOA_SIMCLR = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_etm_toa_simclr-70b5575f.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_etm_toa_simclr-70b5575f.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -319,7 +319,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_ETM_SR_MOCO = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_etm_sr_moco-1266cde3.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_etm_sr_moco-1266cde3.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -332,7 +332,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_ETM_SR_SIMCLR = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_etm_sr_simclr-e5d185d7.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_etm_sr_simclr-e5d185d7.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -345,7 +345,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

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
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_oli_tirs_toa_moco-de7f5e0f.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -358,7 +358,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

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
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_oli_tirs_toa_simclr-030cebfe.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -371,7 +371,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_OLI_SR_MOCO = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_oli_sr_moco-ff580dad.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_oli_sr_moco-ff580dad.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -384,7 +384,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

LANDSAT_OLI_SR_SIMCLR = Weights(
url="https://huggingface.co/torchgeo/ssl4eo_landsat/resolve/main/resnet50_landsat_oli_sr_simclr-94f78913.pth", # noqa: E501
url="https://hf.co/torchgeo/ssl4eo_landsat/resolve/1c88bb51b6e17a21dde5230738fa38b74bd74f76/resnet50_landsat_oli_sr_simclr-94f78913.pth", # noqa: E501
transforms=_ssl4eo_l_transforms,
meta={
"dataset": "SSL4EO-L",
Expand All @@ -397,7 +397,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

SENTINEL1_ALL_MOCO = Weights(
url="https://huggingface.co/torchgeo/resnet50_sentinel1_all_moco/resolve/main/resnet50_sentinel1_all_moco-906e4356.pth", # noqa: E501
url="https://hf.co/torchgeo/resnet50_sentinel1_all_moco/resolve/e79862c667853c10a709bdd77ea8ffbad0e0f1cf/resnet50_sentinel1_all_moco-906e4356.pth", # noqa: E501
transforms=_zhu_xlab_transforms,
meta={
"dataset": "SSL4EO-S12",
Expand All @@ -410,7 +410,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

SENTINEL2_ALL_DINO = Weights(
url="https://huggingface.co/torchgeo/resnet50_sentinel2_all_dino/resolve/main/resnet50_sentinel2_all_dino-d6c330e9.pth", # noqa: E501
url="https://hf.co/torchgeo/resnet50_sentinel2_all_dino/resolve/d7f14bf5530d70ac69d763e58e77e44dbecfec7c/resnet50_sentinel2_all_dino-d6c330e9.pth", # noqa: E501
transforms=_zhu_xlab_transforms,
meta={
"dataset": "SSL4EO-S12",
Expand All @@ -423,7 +423,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

SENTINEL2_ALL_MOCO = Weights(
url="https://huggingface.co/torchgeo/resnet50_sentinel2_all_moco/resolve/main/resnet50_sentinel2_all_moco-df8b932e.pth", # noqa: E501
url="https://hf.co/torchgeo/resnet50_sentinel2_all_moco/resolve/da4f3c9dbe09272eb902f3b37f46635fa4726879/resnet50_sentinel2_all_moco-df8b932e.pth", # noqa: E501
transforms=_zhu_xlab_transforms,
meta={
"dataset": "SSL4EO-S12",
Expand All @@ -436,7 +436,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

SENTINEL2_RGB_MOCO = Weights(
url="https://huggingface.co/torchgeo/resnet50_sentinel2_rgb_moco/resolve/main/resnet50_sentinel2_rgb_moco-2b57ba8b.pth", # noqa: E501
url="https://hf.co/torchgeo/resnet50_sentinel2_rgb_moco/resolve/efd9723b59a88e9dc1420dc1e96afb25b0630a3c/resnet50_sentinel2_rgb_moco-2b57ba8b.pth", # noqa: E501
transforms=_zhu_xlab_transforms,
meta={
"dataset": "SSL4EO-S12",
Expand All @@ -449,7 +449,7 @@ class ResNet50_Weights(WeightsEnum): # type: ignore[misc]
)

SENTINEL2_RGB_SECO = Weights(
url="https://huggingface.co/torchgeo/resnet50_sentinel2_rgb_seco/resolve/main/resnet50_sentinel2_rgb_seco-018bf397.pth", # noqa: E501
url="https://hf.co/torchgeo/resnet50_sentinel2_rgb_seco/resolve/fbd07b02a8edb8fc1035f7957160deed4321c145/resnet50_sentinel2_rgb_seco-018bf397.pth", # noqa: E501
transforms=_seco_transforms,
meta={
"dataset": "SeCo Dataset",
Expand Down
Loading

0 comments on commit d64d9b1

Please sign in to comment.