rslearn 0.0.9__py3-none-any.whl → 0.0.12__py3-none-any.whl

{rslearn-0.0.9.dist-info → rslearn-0.0.12.dist-info}/RECORD

@@ -44,21 +44,21 @@ rslearn/dataset/materialize.py,sha256=-z47svc_JqGhzkp8kq5Hd9fykWNqFEUCQezo887TWB
 rslearn/dataset/remap.py,sha256=6MaImsY02GNACpvRM81RvWmjZWRfAHxo_R3Ox6XLF6A,2723
 rslearn/dataset/window.py,sha256=I5RqZ12jlIXhohw4qews1x_I4tSDpml709DZRtLiN24,12546
 rslearn/models/__init__.py,sha256=_vWoF9d2Slah8-6XhYhdU4SRsy_CNxXjCGQTD2yvu3Q,22
-rslearn/models/anysat.py,sha256=3BnaiS1sYB4SnV6qRjHksiz_r9vUuZeGPUO2XUziFA0,7810
+rslearn/models/anysat.py,sha256=3Oh2gWxicVdUzOjevBEZf0PuolmCy0KC5Ad7JY-0Plc,7949
 rslearn/models/clip.py,sha256=u5aqYnVB4Jag7o1h8EzPDAc1t2BAHeALA9FcUwP5tfo,2238
 rslearn/models/conv.py,sha256=fWyByeswIOKKzyPmP3erYUlZaKEV0huWHA4CyKTBbfY,1703
-rslearn/models/copernicusfm.py,sha256=3AiORuUre9sZYwydbrDgShwKtxeTLmExp7WQmJtBylg,7842
 rslearn/models/croma.py,sha256=cOazTp3l2PNJltKrmPqD5Gy4pi3CI03-X9G4T10cX2k,9529
-rslearn/models/dinov3.py,sha256=GKk5qXZPCEporATJdjaSWsDTfWDlAGRWBplFUJN5nRM,6146
+rslearn/models/dinov3.py,sha256=9k9kNlXCorQQwKjLGptooANd48TUBsITQ1e4fUomlM4,6337
 rslearn/models/faster_rcnn.py,sha256=uaxX6-E1f0BibaA9sorEg3be83C7kTdTc39pC5jRqwE,8286
+rslearn/models/feature_center_crop.py,sha256=24eOrvLEGGVWPw7kPHyUes5HtYNAX7GZ_NpqDGMILEY,1553
 rslearn/models/fpn.py,sha256=s3cz29I14FaSuvBvLOcwCrqVsaRBxG5GjLlqap4WgPc,1603
 rslearn/models/module_wrapper.py,sha256=H2zb-8Au4t31kawW_4JEKHsaXFjpYDawb31ZEauKcxU,2728
 rslearn/models/molmo.py,sha256=mVrARBhZciMzOgOOjGB5AHlPIf2iO9IBSJmdyKSl1L8,2061
 rslearn/models/multitask.py,sha256=j2Kiwj_dUiUp_CIUr25bS8HiyeoFlr1PGqjTfpgIGLc,14672
 rslearn/models/panopticon.py,sha256=woNEs53wVc5D-NxbSDEPRZ_mYe8vllnuldmADjvhfDQ,5806
 rslearn/models/pick_features.py,sha256=y8e4tJFhyG7ZuVSElWhQ5-Aer4ZKJCEH9wLGJU7WqGI,1551
-rslearn/models/pooling_decoder.py,sha256=jZfEQCfthfa21C9sEjgFHUcfhHMVlvG7_nDMw_1FLaE,2727
-rslearn/models/prithvi.py,sha256=SVM3ypJlVTkXQ69pPhB4UeJr87VnmADTCuyV365dbkU,39961
+rslearn/models/pooling_decoder.py,sha256=unr2fSE_QmJHPi3dKtopqMtb1Kn-2h94LgwwAVP9vZg,4437
+rslearn/models/prithvi.py,sha256=AIzcO5xk1ggR0MjbfhIzqPVgUKFN7odxygmgyAelfW8,40143
 rslearn/models/registry.py,sha256=yCcrOvLkbn07Xtln1j7hAB_kmGw0MGsiR2TloJq9Bmk,504
 rslearn/models/resize_features.py,sha256=asKXWrLHIBrU6GaAV0Ory9YuK7IK104XjhkB4ljzI3A,1289
 rslearn/models/sam2_enc.py,sha256=gNlPokr7eNxO2KvnzDMXNxYM2WRO0YkQPjR4110n6cw,3508
@@ -75,14 +75,6 @@ rslearn/models/upsample.py,sha256=3kWbyWZIk56JJxj8en9pieitbrk3XnbIsTKlEkiDQQY,93
 rslearn/models/use_croma.py,sha256=OSBqMuLp-pDtqPNWAVBfmX4wckmyYCKtUDdGCjJk_K8,17966
 rslearn/models/clay/clay.py,sha256=5RO5H8EM0tKjCwWMQ4xDkKkUCwKpm2K_Yw1alnhvVhU,7773
 rslearn/models/clay/configs/metadata.yaml,sha256=rZTFh4Yb9htEfbQNOPl4HTbFogEhzwIRqFzG-1uT01Y,4652
-rslearn/models/copernicusfm_src/__init__.py,sha256=8QLhisbHub6VJl6egijnrOPKK5QNAe5FJhfcxEelj4Y,22
-rslearn/models/copernicusfm_src/dynamic_hypernetwork.py,sha256=aWH5_PgmS8umIwRbGA42RuEx-stb13z1nBjyUhBtaN4,18218
-rslearn/models/copernicusfm_src/model_vit.py,sha256=3coM_xYILlFY2TJiACmQBSe2z16jSG80SVEad_3uB3Q,11396
-rslearn/models/copernicusfm_src/aurora/area.py,sha256=ssg9aXgoZktOsFcEXDEY9670aPUN_PHfCOfDMtpsz1s,1711
-rslearn/models/copernicusfm_src/aurora/fourier.py,sha256=bmoNV3P6CH8R6W2GFuVW8zT_frQVaL-PAgpN3aFS5fA,4414
-rslearn/models/copernicusfm_src/flexivit/patch_embed.py,sha256=EQgbsHBXDq0dTM9kApmmIqd5ZV2X9CPuA_AytbE51uM,9363
-rslearn/models/copernicusfm_src/flexivit/utils.py,sha256=tLBlzgT5bpwMSvyir46bPRWsMmRKh8s7VwMNuvSatGo,2192
-rslearn/models/copernicusfm_src/util/pos_embed.py,sha256=dUYuM_Nch2LB8jQ7UDTmFj36KWe4mM9bsY6dv5m_yZI,8511
 rslearn/models/detr/__init__.py,sha256=GGAnTIhyuvl34IRrJ_4gXjm_01OlM5rbQQ3c3TGfbK8,84
 rslearn/models/detr/box_ops.py,sha256=ORCF6EwMpMBB_VgQT05SjR47dCR2rN2gPhL_gsuUWJs,3236
 rslearn/models/detr/detr.py,sha256=otLmmyUm05e4MUyvQBoqo-RKnx3hbodTXvfPQWvuTEI,18737
@@ -93,6 +85,9 @@ rslearn/models/detr/util.py,sha256=NMHhHbkIo7PoBUVbDqa2ZknJBTswmaxFCGHrPtFKnGg,6
 rslearn/models/galileo/__init__.py,sha256=QQa0C29nuPRva0KtGiMHQ2ZB02n9SSwj_wqTKPz18NM,112
 rslearn/models/galileo/galileo.py,sha256=jUHA64YvVC3Fz5fevc_9dFJfZaINODRDrhSGLIiOZcw,21115
 rslearn/models/galileo/single_file_galileo.py,sha256=l5tlmmdr2eieHNH-M7rVIvcptkv0Fuk3vKXFW691ezA,56143
+rslearn/models/olmoearth_pretrain/__init__.py,sha256=AjRvbjBdadCdPh-EdvySH76sVAQ8NGQaJt11Tsn1D5I,36
+rslearn/models/olmoearth_pretrain/model.py,sha256=I_RWFbwzO5yCWpEcEQP8PeiD8M1QpeMtVrjl15evIHU,10632
+rslearn/models/olmoearth_pretrain/norm.py,sha256=rHjFyWkpNLYMx9Ow7TsU-jGm9Sjx7FVf0p4R__ohx2c,3266
 rslearn/models/panopticon_data/sensors/drone.yaml,sha256=xqWS-_QMtJyRoWXJm-igoSur9hAmCFdqkPin8DT5qpw,431
 rslearn/models/panopticon_data/sensors/enmap.yaml,sha256=b2j6bSgYR2yKR9DRm3SPIzSVYlHf51ny_p-1B4B9sB4,13431
 rslearn/models/panopticon_data/sensors/goes.yaml,sha256=o00aoWCYqam0aB1rPmXq1MKe8hsKak_qyBG7BPL27Sc,152
@@ -114,7 +109,7 @@ rslearn/tile_stores/tile_store.py,sha256=9AeYduDYPp_Ia2NMlq6osptpz_AFGIOQcLJrqZ_
 rslearn/train/__init__.py,sha256=fnJyY4aHs5zQqbDKSfXsJZXY_M9fbTsf7dRYaPwZr2M,30
 rslearn/train/data_module.py,sha256=K-nQgnOZn-KGq_G2pVOQFtWRrlWih0212i_bkXZ2bEE,23515
 rslearn/train/dataset.py,sha256=YiskNlYYcKqZxyw0Xzop1RGLbjMc-oK_rmhrSMVbTQg,51857
-rslearn/train/lightning_module.py,sha256=ge2z8trU7cMvxBeqUXC1tB44pftzitw7DRsIa6asBS4,14623
+rslearn/train/lightning_module.py,sha256=ZLBiId3secUlVs2yzkN-mwVv4rMdh5TkdZYl4vv_Cw0,14466
 rslearn/train/optimizer.py,sha256=EKSqkmERalDA0bF32Gey7n6z69KLyaUWKlRsGJfKBmE,927
 rslearn/train/prediction_writer.py,sha256=YNs92QqPrqbREZXoE-aPa_oKQW0C9LvZAY129vyvI08,13288
 rslearn/train/scheduler.py,sha256=wFbmycMHgL6nRYeYalDjb0G8YVo8VD3T3sABS61jJ7c,2318
@@ -124,11 +119,11 @@ rslearn/train/callbacks/freeze_unfreeze.py,sha256=8fIzBMhCKKjpTffIeAdhdSjsBd8NjT
 rslearn/train/callbacks/gradients.py,sha256=4YqCf0tBb6E5FnyFYbveXfQFlgNPyxIXb2FCWX4-6qs,5075
 rslearn/train/callbacks/peft.py,sha256=wEOKsS3RhsRaZTXn_Kz2wdsZdIiIaZPdCJWtdJBurT8,4156
 rslearn/train/tasks/__init__.py,sha256=dag1u72x1-me6y0YcOubUo5MYZ0Tjf6-dOir9UeFNMs,75
-rslearn/train/tasks/classification.py,sha256=DI0_Wzs-9rNPWokvfxi1BIA6QyqNee42SpptQx82WHM,13182
-rslearn/train/tasks/detection.py,sha256=OoZzC8ZbmhyZ30tD-4cB-3Jj0AN6Y7hg0wk27rDguCE,22297
+rslearn/train/tasks/classification.py,sha256=kahVdXPU6fDwDCdqlrjZGb9uA-PYG74DbQQ0kJUt-Eg,13186
+rslearn/train/tasks/detection.py,sha256=9j9webusrjGexvUmZ7gl3NTBS63Qq511VFlB2WbLi5Y,22302
 rslearn/train/tasks/multi_task.py,sha256=dBWsnbvQ0CReNsbDHmZ_-sXjUE0H4S2OPcbJwMquG9g,6016
-rslearn/train/tasks/per_pixel_regression.py,sha256=tkVntKFzPlWFxdupPlMfhIRWlJ0UCgxg_FGhcA2-wjE,8649
-rslearn/train/tasks/regression.py,sha256=_PoxOfWNseujD4IWsuTL82fAAXgtco4WdfkNXQ68Nbg,11497
+rslearn/train/tasks/per_pixel_regression.py,sha256=W8dbLyIiPgFI3gA_aZQX0pSFRWLP2v6tthsFbKhcDVg,8783
+rslearn/train/tasks/regression.py,sha256=zZhrrZ1qxjrdLjKWC9McRivDXCcKiYfdLC-kaMeVkDc,11547
 rslearn/train/tasks/segmentation.py,sha256=xEni3CLDyetviv84XrpJg5xeJU87WHGFKTVfIeemGIY,21868
 rslearn/train/tasks/task.py,sha256=4w2xKL_U5JAtdj2dYoVv82h6xTtgUsA3IvIOcXyZecs,3887
 rslearn/train/transforms/__init__.py,sha256=BkCAzm4f-8TEhPIuyvCj7eJGh36aMkZFYlq-H_jkSvY,778
@@ -137,7 +132,7 @@ rslearn/train/transforms/crop.py,sha256=4jA3JJsC0ghicPHbfsNJ0d3WpChyvftY73ONiwQa
 rslearn/train/transforms/flip.py,sha256=lkTeje3T8gNn2gt6957morXq1fGNho-apSpCvNp0_9o,3480
 rslearn/train/transforms/mask.py,sha256=pwt33XXWLwldLiar-PgVgBQzQd1qfL18SPz3LYQMoYM,2111
 rslearn/train/transforms/normalize.py,sha256=uyv2hE5hw5B2kCRHa4JIx0tfowm-C7bgumwINvvfyts,5014
-rslearn/train/transforms/pad.py,sha256=EDswS9KYRSloM3DQlbCz6S0WYqFQJvI433qMqTtqrZw,4686
+rslearn/train/transforms/pad.py,sha256=pj4Ql8GSRrhg8KOZTNPB40Qq8CoCCHdGo04uficik84,4698
 rslearn/train/transforms/select_bands.py,sha256=uDfD9G8Z4VTt88QZsjj1FB20QEmzSefhKf7uDXYn77M,2441
 rslearn/train/transforms/sentinel1.py,sha256=FrLaYZs2AjqWQCun8DTFtgo1l0xLxqaFKtDNIehtpDg,1913
 rslearn/train/transforms/transform.py,sha256=n1Qzqix2dVvej-Q7iPzHeOQbqH79IBlvqPoymxhNVpE,4446
@@ -156,9 +151,10 @@ rslearn/utils/spatial_index.py,sha256=eomJAUgzmjir8j9HZnSgQoJHwN9H0wGTjmJkMkLLfs
 rslearn/utils/sqlite_index.py,sha256=YGOJi66544e6JNtfSft6YIlHklFdSJO2duxQ4TJ2iu4,2920
 rslearn/utils/time.py,sha256=2ilSLG94_sxLP3y5RSV5L5CG8CoND_dbdzYEHVtN-I8,387
 rslearn/utils/vector_format.py,sha256=EIChYCL6GLOILS2TO2JBkca1TuaWsSubWv6iRS3P2ds,16139
-rslearn-0.0.9.dist-info/licenses/LICENSE,sha256=_99ZWPoLdlUbqZoSC5DF4ihiNwl5rTEmBaq2fACecdg,11352
-rslearn-0.0.9.dist-info/METADATA,sha256=6BV8wt9tuo94FkoKjR3RcF3AbKNbU3IodkJtK4tASkE,36248
-rslearn-0.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rslearn-0.0.9.dist-info/entry_points.txt,sha256=doTBQ57NT7nq-dgYGgTTw6mafcGWb_4PWYtYR4rGm50,46
-rslearn-0.0.9.dist-info/top_level.txt,sha256=XDKo90WBH8P9RQumHxo0giLJsoufT4r9odv-WE6Ahk4,8
-rslearn-0.0.9.dist-info/RECORD,,
+rslearn-0.0.12.dist-info/licenses/LICENSE,sha256=_99ZWPoLdlUbqZoSC5DF4ihiNwl5rTEmBaq2fACecdg,11352
+rslearn-0.0.12.dist-info/licenses/NOTICE,sha256=wLPr6rwV_jCg-xEknNGwhnkfRfuoOE9MZ-lru2yZyLI,5070
+rslearn-0.0.12.dist-info/METADATA,sha256=0jHeiz1QCT56zOws1CGGFVM9TotMOWIboQmGASdZAwY,36318
+rslearn-0.0.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rslearn-0.0.12.dist-info/entry_points.txt,sha256=doTBQ57NT7nq-dgYGgTTw6mafcGWb_4PWYtYR4rGm50,46
+rslearn-0.0.12.dist-info/top_level.txt,sha256=XDKo90WBH8P9RQumHxo0giLJsoufT4r9odv-WE6Ahk4,8
+rslearn-0.0.12.dist-info/RECORD,,

rslearn-0.0.12.dist-info/licenses/NOTICE

@@ -0,0 +1,115 @@
+rslearn is released under Apache License 2.0
+Copyright 2025 Allen Institute for AI
+
+The following third party code is included in this repository.
+
+====================
+
+rslearn.models.detr is adapted from https://github.com/facebookresearch/detr which is
+released under Apache License 2.0.
+
+Copyright 2020 - present, Facebook, Inc
+
+====================
+
+rslearn.models.use_croma is copied from https://github.com/antofuller/CROMA
+
+MIT License
+
+Copyright (c) 2023 Anthony Fuller
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+
+====================
+
+rslearn.models.galileo is adapted from https://github.com/nasaharvest/galileo
+
+MIT License
+
+Copyright (c) 2024 Presto Authors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+
+====================
+
+rslearn.models.presto is adapted from https://github.com/nasaharvest/presto
+
+MIT License
+
+Copyright (c) 2024 Presto Authors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+
+====================
+
+rslearn.models.prithvi includes code adapted from https://github.com/NASA-IMPACT/Prithvi-WxC
+
+MIT License
+
+Copyright (c) 2024 Inter Agency Implementation and Advanced Concepts
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

rslearn/models/copernicusfm.py

@@ -1,228 +0,0 @@
-"""Copernicus FM model."""
-
-import logging
-import math
-from enum import Enum
-from pathlib import Path
-
-import torch
-import torch.nn.functional as F
-from einops import rearrange
-from huggingface_hub import hf_hub_download
-
-from .copernicusfm_src.model_vit import vit_base_patch16
-
-logger = logging.getLogger(__name__)
-
-
-class CopernicusFMModality(Enum):
-    """Modality for Copernicus FM."""
-
-    SENTINEL2_L2A = "sentinel2_l2a"
-    SENTINEL1 = "sentinel1"
-
-
-MODALITY_TO_WAVELENGTH_BANDWIDTHS: dict[str, dict[str, list]] = {
-    # https://github.com/zhu-xlab/Copernicus-FM/blob/main/Copernicus-Bench/src/configs/dataset/cobench_eurosat_s2.yaml
-    CopernicusFMModality.SENTINEL2_L2A.value: {
-        "band_names": [
-            "B01",
-            "B02",
-            "B03",
-            "B04",
-            "B05",
-            "B06",
-            "B07",
-            "B08",
-            "B8A",
-            "B09",
-            "B10",
-            "B11",
-            "B12",
-        ],
-        "band_wavelengths": [
-            440,
-            490,
-            560,
-            665,
-            705,
-            740,
-            783,
-            842,
-            860,
-            940,
-            1370,
-            1610,
-            2190,
-        ],
-        "band_bandwidths": [20, 65, 35, 30, 15, 15, 20, 115, 20, 20, 30, 90, 180],
-    },
-    # https://github.com/zhu-xlab/Copernicus-FM/blob/main/Copernicus-Bench/src/configs/dataset/cobench_eurosat_s1.yaml
-    CopernicusFMModality.SENTINEL1.value: {
-        "band_names": ["vv", "vh"],
-        "band_wavelengths": [50000000, 50000000],
-        "band_bandwidths": [1e9, 1e9],
-    },
-}
-
-HF_REPO_ID = "wangyi111/Copernicus-FM"
-HF_REPO_REVISION = "e1db406d517a122c8373802e1c130c5fc4789f84"
-HF_FILENAME = "CopernicusFM_ViT_base_varlang_e100.pth"
-
-
-class CopernicusFM(torch.nn.Module):
-    """Wrapper for Copernicus FM to ingest Masked Helios Sample."""
-
-    image_resolution = 224
-    patch_size = 16
-    input_mode = "spectral"
-    # Don't need this as band order is provided
-    supported_modalities = [
-        CopernicusFMModality.SENTINEL2_L2A.value,
-        CopernicusFMModality.SENTINEL1.value,
-    ]
-
-    def __init__(
-        self,
-        band_order: dict[str, list[str]],
-        cache_dir: str | Path | None = None,
-    ) -> None:
-        """Initialize the Copernicus FM wrapper.
-
-        Args:
-            band_order: The band order for each modality that will be used. The bands
-                can be provided in any order, and any subset can be used.
-            cache_dir: The directory to cache the weights. If None, a default directory
-                managed by huggingface_hub is used. The weights are downloaded from
-                Hugging Face (https://huggingface.co/wangyi111/Copernicus-FM).
-        """
-        super().__init__()
-
-        # Make sure all keys in band_order are in supported_modalities.
-        for modality_name in band_order.keys():
-            if modality_name in self.supported_modalities:
-                continue
-            raise ValueError(
-                f"band_order contains unsupported modality {modality_name}"
-            )
-
-        # global_pool=True so that we initialize the fc_norm layer
-        self.model = vit_base_patch16(num_classes=10, global_pool=True)
-
-        # Load weights, downloading if needed.
-        local_fname = hf_hub_download(
-            repo_id=HF_REPO_ID,
-            revision=HF_REPO_REVISION,
-            filename=HF_FILENAME,
-            local_dir=cache_dir,
-        )  # nosec
-        state_dict = torch.load(local_fname, weights_only=True)
-        self.model.load_state_dict(state_dict, strict=False)
-
-        # take MODALITY_TO_WAVELENGTH_BANDWIDTHS and rearrange it so that it has the same
-        # ordering as the user-provided band order.
-        self.modality_to_wavelength_bandwidths = {}
-        for modality in self.supported_modalities:
-            if modality not in band_order:
-                continue
-
-            wavelength_bandwidths = MODALITY_TO_WAVELENGTH_BANDWIDTHS[modality]
-            wavelengths = []
-            bandwidths = []
-            for b in band_order[modality]:
-                cfm_idx = wavelength_bandwidths["band_names"].index(b)
-                wavelengths.append(wavelength_bandwidths["band_wavelengths"][cfm_idx])
-                bandwidths.append(wavelength_bandwidths["band_bandwidths"][cfm_idx])
-            self.modality_to_wavelength_bandwidths[modality] = {
-                "band_bandwidths": bandwidths,
-                "band_wavelengths": wavelengths,
-            }
-
-    def _resize_data(self, data: torch.Tensor) -> torch.Tensor:
-        """Process individual modality data.
-
-        Args:
-            data: Input tensor of shape [B, C, H, W]
-
-        Returns:
-            list of tensors of shape [B, C, H, W]
-        """
-        # Get original dimensions
-        original_height = data.shape[2]
-        new_height = self.patch_size if original_height == 1 else self.image_resolution
-        data = F.interpolate(
-            data,
-            size=(new_height, new_height),
-            mode="bilinear",
-            align_corners=False,
-        )
-        return data
-
-    def prepare_input(
-        self,
-        inputs: dict[str, torch.Tensor],
-    ) -> tuple[torch.Tensor, list[int], list[int]]:
-        """Prepare input for the CopernicusFM model from MaskedHeliosSample."""
-        wavelengths: list[int] = []
-        bandwidths: list[int] = []
-        all_processed_data: list[list[torch.Tensor]] = []
-        for modality in inputs.keys():
-            if modality not in self.supported_modalities:
-                logger.debug(
-                    f"Skipping modality {modality} as it is not in the supported "
-                    f"modalities list {self.supported_modalities}"
-                )
-                continue
-
-            data = inputs[modality]
-
-            if data is None:
-                continue
-
-            all_processed_data.append(self._resize_data(data))
-            wavelengths.extend(
-                self.modality_to_wavelength_bandwidths[modality]["band_wavelengths"]
-            )
-            bandwidths.extend(
-                self.modality_to_wavelength_bandwidths[modality]["band_bandwidths"]
-            )
-
-        concatenated_processed_data = torch.cat(all_processed_data, dim=1)
-        return concatenated_processed_data, wavelengths, bandwidths
-
-    def forward(
-        self,
-        inputs: list[dict[str, torch.Tensor]],
-    ) -> torch.Tensor:
-        """Forward pass through CopernicusFM model."""
-        batch_inputs = {
-            key: torch.stack([inp[key] for inp in inputs], dim=0)
-            for key in inputs[0].keys()
-        }
-        # Prepare input
-        data, wavelengths, bandwidths = self.prepare_input(batch_inputs)
-        meta = torch.full(
-            (1, 4), float("nan"), device=data.device
-        )  # [lon, lat, delta_time, patch_token_area], assume unknown
-        # "The embed tensor contains the encoded image features, which can be used for downstream tasks."
-        _, timestep_output = self.model(
-            data,
-            meta,
-            wavelengths,
-            bandwidths,
-            None,
-            self.input_mode,
-            self.patch_size,
-        )
-        # no norm, following
-        # https://github.com/zhu-xlab/Copernicus-FM/blob/main/Copernicus-Bench/src/foundation_models/CopernicusFM/models_dwv_seg.py
-        side = math.isqrt(timestep_output.shape[1])
-        output_features = rearrange(
-            timestep_output, "b (h w) c -> b c h w ", h=side, w=side
-        )
-        return [output_features]
-
-    def get_backbone_channels(self) -> list[tuple[int, int]]:
-        """Returns the output channels of this model when used as a backbone."""
-        # TODO: load this from a constant depending on the model size
-        return [(self.patch_size, 768)]

rslearn/models/copernicusfm_src/__init__.py

@@ -1 +0,0 @@
-# mypy: ignore-errors

rslearn/models/copernicusfm_src/aurora/area.py

@@ -1,50 +0,0 @@
-"""Copyright (c) Microsoft Corporation. Licensed under the MIT license."""
-
-import torch
-
-__all__ = ["area", "radius_earth"]
-
-
-# float: Radius of the earth in kilometers.
-radius_earth = 6378137 / 1000
-
-
-def area(polygon: torch.Tensor) -> torch.Tensor:
-    """Compute the area of a polygon specified by latitudes and longitudes in degrees.
-
-    This function is a PyTorch port of the PyPI package `area`. In particular, it is heavily
-    inspired by the following file:
-
-        https://github.com/scisco/area/blob/9d9549d6ebffcbe4bffe11b71efa2d406d1c9fe9/area/__init__.py
-
-    Args:
-        polygon (:class:`torch.Tensor`): Polygon of the shape `(*b, n, 2)` where `b` is an optional
-            multidimensional batch size, `n` is the number of points of the polygon, and 2
-            concatenates first latitudes and then longitudes. The polygon does not have be closed.
-
-    Returns:
-        :class:`torch.Tensor`: Area in square kilometers.
-    """
-    # Be sure to close the loop.
-    polygon = torch.cat((polygon, polygon[..., -1:, :]), axis=-2)
-
-    area = torch.zeros(polygon.shape[:-2], dtype=polygon.dtype, device=polygon.device)
-    n = polygon.shape[-2]  # Number of points of the polygon
-
-    rad = torch.deg2rad  # Convert degrees to radians.
-
-    if n > 2:
-        for i in range(n):
-            i_lower = i
-            i_middle = (i + 1) % n
-            i_upper = (i + 2) % n
-
-            lon_lower = polygon[..., i_lower, 1]
-            lat_middle = polygon[..., i_middle, 0]
-            lon_upper = polygon[..., i_upper, 1]
-
-            area = area + (rad(lon_upper) - rad(lon_lower)) * torch.sin(rad(lat_middle))
-
-    area = area * radius_earth * radius_earth / 2
-
-    return torch.abs(area)

rslearn/models/copernicusfm_src/aurora/fourier.py

@@ -1,134 +0,0 @@
-# type: ignore
-"""Copyright (c) Microsoft Corporation. Licensed under the MIT license."""
-
-import math
-
-import numpy as np
-import torch
-import torch.nn as nn
-
-from .area import area, radius_earth
-
-__all__ = [
-    "FourierExpansion",
-    "pos_expansion",
-    "scale_expansion",
-    "lead_time_expansion",
-    "levels_expansion",
-    "absolute_time_expansion",
-]
-
-
-class FourierExpansion(nn.Module):
-    """A Fourier series-style expansion into a high-dimensional space.
-
-    Attributes:
-        lower (float): Lower wavelength.
-        upper (float): Upper wavelength.
-        assert_range (bool): Assert that the encoded tensor is within the specified wavelength
-            range.
-    """
-
-    def __init__(self, lower: float, upper: float, assert_range: bool = True) -> None:
-        """Initialise.
-
-        Args:
-            lower (float): Lower wavelength.
-            upper (float): Upper wavelength.
-            assert_range (bool, optional): Assert that the encoded tensor is within the specified
-                wavelength range. Defaults to `True`.
-        """
-        super().__init__()
-        self.lower = lower
-        self.upper = upper
-        self.assert_range = assert_range
-
-    def forward(self, x: torch.Tensor, d: int) -> torch.Tensor:
-        """Perform the expansion.
-
-        Adds a dimension of length `d` to the end of the shape of `x`.
-
-        Args:
-            x (:class:`torch.Tensor`): Input to expand of shape `(..., n)`. All elements of `x` must
-                lie within `[self.lower, self.upper]` if `self.assert_range` is `True`.
-            d (int): Dimensionality. Must be a multiple of two.
-
-        Raises:
-            AssertionError: If `self.assert_range` is `True` and not all elements of `x` are not
-                within `[self.lower, self.upper]`.
-            ValueError: If `d` is not a multiple of two.
-
-        Returns:
-            torch.Tensor: Fourier series-style expansion of `x` of shape `(..., n, d)`.
-        """
-        # If the input is not within the configured range, the embedding might be ambiguous!
-        in_range = torch.logical_and(
-            self.lower <= x.abs(), torch.all(x.abs() <= self.upper)
-        )
-        in_range_or_zero = torch.all(
-            torch.logical_or(in_range, x == 0)
-        )  # Allow zeros to pass through.
-        if self.assert_range and not in_range_or_zero:
-            raise AssertionError(
-                f"The input tensor is not within the configured range"
-                f" `[{self.lower}, {self.upper}]`."
-            )
-
-        # We will use half of the dimensionality for `sin` and the other half for `cos`.
-        if not (d % 2 == 0):
-            raise ValueError("The dimensionality must be a multiple of two.")
-
-        # Always perform the expansion with `float64`s to avoid numerical accuracy shenanigans.
-        x = x.double()
-
-        wavelengths = torch.logspace(
-            math.log10(self.lower),
-            math.log10(self.upper),
-            d // 2,
-            base=10,
-            device=x.device,
-            dtype=x.dtype,
-        )
-        prod = torch.einsum("...i,j->...ij", x, 2 * np.pi / wavelengths)
-        encoding = torch.cat((torch.sin(prod), torch.cos(prod)), dim=-1)
-
-        return encoding.float()  # Cast to `float32` to avoid incompatibilities.
-
-
-# Determine a reasonable smallest value for the scale embedding by assuming a smallest delta in
-# latitudes and longitudes.
-_delta = 0.01  # Reasonable smallest delta in latitude and longitude
-_min_patch_area: float = area(
-    torch.tensor(
-        [
-            # The smallest patches will be at the poles. Just use the north pole.
-            [90, 0],
-            [90, _delta],
-            [90 - _delta, _delta],
-            [90 - _delta, 0],
-        ],
-        dtype=torch.float64,
-    )
-).item()
-_area_earth = 4 * np.pi * radius_earth * radius_earth
-
-pos_expansion = FourierExpansion(_delta, 720)
-
-
-scale_expansion = FourierExpansion(_min_patch_area, _area_earth)
-
-
-lead_time_expansion = FourierExpansion(1 / 60, 24 * 7 * 3)
-
-levels_expansion = FourierExpansion(0.01, 1e5)
-
-absolute_time_expansion = FourierExpansion(1, 24 * 365.25, assert_range=False)
-
-### new for SSL4EO-S ###
-# min wavelength: ultraviolet light (100 nm)
-# max wavelength: radio waves (1 m)
-spectrum_central_expansion = FourierExpansion(1e-7, 1)
-
-# min bandwidth: 10nm
-# max bandwidth: 1m
-spectrum_width_expansion = FourierExpansion(1e-7, 1)