rslearn 0.0.1__py3-none-any.whl → 0.0.21__py3-none-any.whl

rslearn-0.0.21.dist-info/RECORD

@@ -0,0 +1,167 @@
+rslearn/__init__.py,sha256=fFmAen3vxZyosEfPbG0W46IttujYGVxzrGkJ0YutmmY,73
+rslearn/arg_parser.py,sha256=Go1MyEflcau_cziirmNd7Yhxa0WtXTAljIVE4f5H1GE,1194
+rslearn/const.py,sha256=FUCfsvFAs-QarEDJ0grdy0C1HjUjLpNFYGo5I2Vpc5Y,449
+rslearn/lightning_cli.py,sha256=1eTeffUlFqBe2KnyuYyXJdNKYQClCA-PV1xr0vJyJao,17972
+rslearn/log_utils.py,sha256=unD9gShiuO7cx5Nnq8qqVQ4qrbOOwFVgcHxN5bXuiAo,941
+rslearn/main.py,sha256=jRMYeU3-QvYSkTAJB69S1mHEft7-5_-RomzX1B-b8GM,28581
+rslearn/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rslearn/template_params.py,sha256=Vop0Ha-S44ctCa9lvSZRjrMETznJZlR5y_gJrVIwrPg,791
+rslearn/config/__init__.py,sha256=n1qpZ0ImshTtLYl5mC73BORYyUcjPJyHiyZkqUY1hiY,474
+rslearn/config/dataset.py,sha256=AJlS-41B70E2Y4ST545J2P3Kz83jrFCRdkZecKZ7sQY,23255
+rslearn/data_sources/__init__.py,sha256=zzuZUxrlEIw84YpD2I0HJvCoLDB29LbmnKTXiJykzGU,660
+rslearn/data_sources/aws_landsat.py,sha256=0ZQtmd2NCnvLy4vFSB1AlmoguJbiQB_e_T4eS1tnW9Q,20443
+rslearn/data_sources/aws_open_data.py,sha256=lrHnMJTH3NAaRdNjxwCIxSq8rq90IvV4ho-qAG6Hdgc,29348
+rslearn/data_sources/aws_sentinel1.py,sha256=knBg2ZdwzGRIibUPAqnhYR-DbHqFL4tLsuRVrucTWU4,4745
+rslearn/data_sources/climate_data_store.py,sha256=gwlbugwAT0YoOINOKNVlZuzpxUO5G0tfzwy4XVlURFg,18360
+rslearn/data_sources/copernicus.py,sha256=DLyXwnkFo2LzbxfLKHkOvHyZcNYFg5w-yXQPeBL67_w,35049
+rslearn/data_sources/data_source.py,sha256=lxdcxefETX5ui5uUdZn-ACSBhy6YVo4XDfk368EMD40,4862
+rslearn/data_sources/earthdaily.py,sha256=BNP7BK_vb9yQw6LaIaJ80vjCfBZ9TeALGkjHYIet_W0,18205
+rslearn/data_sources/earthdata_srtm.py,sha256=bwo8e_y9fFPliZ411tTWOiAUlEcb3AWBkeZxnkFY5SI,10633
+rslearn/data_sources/eurocrops.py,sha256=FQqdBcNl43fArq6rd_-iffVJyliIDbB0lIHvNdmtQBU,8663
+rslearn/data_sources/gcp_public_data.py,sha256=E3BhJqOgHMqwcfaxg7D47KNmfAwJDiDjH8-OdqNdjuI,36524
+rslearn/data_sources/google_earth_engine.py,sha256=xdoSDjSVp6lPVPMv4UJZ6BRUozUA2hFbSTl1707TBoM,23523
+rslearn/data_sources/local_files.py,sha256=mo5W_BxBl89EPTIHNDEpXM6qBjrP225KK0PcmNgvJZQ,19090
+rslearn/data_sources/openstreetmap.py,sha256=TzZfouc2Z4_xjx2v_uv7aPn4tVW3flRVQN4qBfl507E,18161
+rslearn/data_sources/planet.py,sha256=6FWQ0bl1k3jwvwp4EVGi2qs3OD1QhnKOKP36mN4HELI,9446
+rslearn/data_sources/planet_basemap.py,sha256=e9R6FlagJjg8Z6Rc1dC6zK3xMkCohz8eohXqXmd29xg,9670
+rslearn/data_sources/planetary_computer.py,sha256=qgvGe_hQDxAIrRVB9szvJZL5IvE2jTo8UWuKCMzx7jM,29773
+rslearn/data_sources/usda_cdl.py,sha256=_WvxZkm0fbXfniRs6NT8iVCbTTmVPflDhsFT2ci6_Dk,6879
+rslearn/data_sources/usgs_landsat.py,sha256=kPOb3hsZe5-guUcFZZkwzcRpYZ3Zo7Bk4E829q_xiyU,18516
+rslearn/data_sources/utils.py,sha256=v_90ALOuts7RHNcx-j8o-aQ_aFjh8ZhXrmsaa9uEGDA,11651
+rslearn/data_sources/vector_source.py,sha256=NCa7CxIrGKe9yRT0NyyFKFQboDGDZ1h7663PV9OfMOM,44
+rslearn/data_sources/worldcereal.py,sha256=rGGxwJAC3pAASVBQoUhwS3qL-qcmF6W__Tb53qWCEmE,21501
+rslearn/data_sources/worldcover.py,sha256=n7gi-JRytxkvkUhKT--dVziMcWSSyMbZA7ZCzLT2MJY,6037
+rslearn/data_sources/worldpop.py,sha256=S3RSc5kTwSs2bcREjNarBsqf3MBX5CN0eHj7Qkx4K74,5625
+rslearn/data_sources/xyz_tiles.py,sha256=P601CvUmoVDC_ZRVhPKaIoPYYCq5UhZ-v7DaGlN5y_0,13797
+rslearn/dataset/__init__.py,sha256=bHtBlEEBCekO-gaJqiww0-VjvZTE5ahx0llleo8bfP8,289
+rslearn/dataset/add_windows.py,sha256=NwIvku6zxCJ9kgVFa5phJc0Gj1Y1bCzh6TLb9nEGl0s,8462
+rslearn/dataset/dataset.py,sha256=YyGFy_VGUaOPfrEQqBl0Fp5JAsH3KNCqo6ZTu3TW4yY,3223
+rslearn/dataset/handler_summaries.py,sha256=wI99RDk5erCWkzl1A7Uc4chatQ9KWIr4F_0Hxr9Co6s,2607
+rslearn/dataset/manage.py,sha256=-lGSIgk6Z7-verF_POwe4n5w9eSpgyt4nEOcOj382rc,18971
+rslearn/dataset/materialize.py,sha256=o05OeLk_wWEOsw15oc5yjpD4J-twGCTfXAtxyAQsQ9I,20974
+rslearn/dataset/remap.py,sha256=6MaImsY02GNACpvRM81RvWmjZWRfAHxo_R3Ox6XLF6A,2723
+rslearn/dataset/window.py,sha256=X4q8YzcSOTtwKxCPf71QLMoyKUtYMSnZu0kPnmVSUx4,10644
+rslearn/dataset/storage/__init__.py,sha256=R50AVV5LH2g7ol0-jyvGcB390VsclXGbJXz4fmkn9as,52
+rslearn/dataset/storage/file.py,sha256=g9HZ3CD4QcgyVNsBaXhjIKQgDOAeZ4R08sJ7ntx4wo8,6815
+rslearn/dataset/storage/storage.py,sha256=DxZ7iwV938PiLwdQzb5EXSb4Mj8bRGmOTmA9fzq_Ge8,4840
+rslearn/models/__init__.py,sha256=_vWoF9d2Slah8-6XhYhdU4SRsy_CNxXjCGQTD2yvu3Q,22
+rslearn/models/anysat.py,sha256=nzk6hB83ltNFNXYRNA1rTvq2AQcAhwyvgBaZui1M37o,8107
+rslearn/models/attention_pooling.py,sha256=VeRoBLk0326Kpj780Pw8EiSFMU5_K9qg6HQO-B2r2PU,7044
+rslearn/models/clip.py,sha256=QG1oUFqcVuNEhx7BNfJ1FnxIOMNUwRNBwXCe3CR6wFI,2415
+rslearn/models/component.py,sha256=uikFDzPYaW_LSXsrSsES1aup4IDIuWHsitWLpKgF7zU,3432
+rslearn/models/concatenate_features.py,sha256=Attemr5KurxlOojpclD0Pd5Cu2KHpNdpXe8jCSjpJ9U,3818
+rslearn/models/conv.py,sha256=dEAAfhPo4bFlZPSAQjzqZTpP-hdJ394TytYssVK-fDA,2001
+rslearn/models/croma.py,sha256=BcKV4-D4uira6f9zvW73aslF_XitAhObnyrE_3iTcTs,11008
+rslearn/models/dinov3.py,sha256=-hyhkxbjns66s9xKT3bPU1dTzanPPc7qJ0dNAwV19-E,6497
+rslearn/models/faster_rcnn.py,sha256=yOipLPmVHbadvYCR9xfCYgmkU9Mot6fgDK-kKicVTlo,8685
+rslearn/models/feature_center_crop.py,sha256=_Mu3E4iJLBug9I4ZIBIpB_VJo-xGterHmhtIFGaHR34,1808
+rslearn/models/fpn.py,sha256=qm7nKMgsZrCoAdz8ASmNKU2nvZ6USm5CedMfy_w_gwE,2079
+rslearn/models/module_wrapper.py,sha256=73JspaglnNabUGZB2EiCYF_dZ3-Kicg_OpoTfUWHONk,2271
+rslearn/models/molmo.py,sha256=lXnevwTCNyc1XcnJUB5_pK1G2AJGYMvQYU21mZFf5u0,2246
+rslearn/models/multitask.py,sha256=bpFxvtFowRyT-tvRSdY7AKbEx_i1y7sToEzZgTMcF4s,16264
+rslearn/models/panopticon.py,sha256=lXXBusXZfwdf10rBVPAQSbaGOMyKCDeEBmXShzvfMoU,5947
+rslearn/models/pick_features.py,sha256=fI9SYubqpCWOAHYGVUSg5sgD31dsnAR9mNuLmqfIeL8,1110
+rslearn/models/pooling_decoder.py,sha256=zrMH6wUExCa-XD1q9CIFD2ScgiasapyJs9plhcUxhIs,4767
+rslearn/models/prithvi.py,sha256=J45eC1pd4l5AGlr19Qjrjrw5PPwvYE9bNM5qCFoznmg,40390
+rslearn/models/resize_features.py,sha256=U7ZIVwwToJJnwchFG59wLWWP9eikHDB_1c4OtpubxHU,1693
+rslearn/models/sam2_enc.py,sha256=WZOtlp0FjaVztW4gEVIcsFQdKArS9iblRODP0b6Oc8M,3641
+rslearn/models/satlaspretrain.py,sha256=2R48ulbtd44Qy2FYJCkllE2Wk35eZxkc79ruSgkmgcQ,3384
+rslearn/models/simple_time_series.py,sha256=Nfk5E3d9W-4AyLQiy-P8p-JvxmFYE3FBrvOgttjXSMw,14678
+rslearn/models/singletask.py,sha256=9DM9a9-Mv3vVQqRhPOIXG2HHuVqVa_zuvgafeeYh4r0,1903
+rslearn/models/ssl4eo_s12.py,sha256=DOlpIj6NfjIlWyJ27m9Xo8TMlovBDstFq0ARnmAJ6qY,3919
+rslearn/models/swin.py,sha256=Xqr3SswbHP6IhwT2atZMAPF2TUzQqfxvihksb8WSeRo,6065
+rslearn/models/task_embedding.py,sha256=Z6sf61BLCtvdrdnvjh8500b-KiFp3GeWbT4mOqpaCKk,9100
+rslearn/models/terramind.py,sha256=Ob0bmpo2YdFb__cWcIdHhyvllNfbZ8FF3dpermutqjQ,8790
+rslearn/models/trunk.py,sha256=1GCH9iyLIytoHVntLSMwfH9duQpe1W4DPmOClLpPKjc,4778
+rslearn/models/unet.py,sha256=HuuINvkB1-5w9ZOTXZCWkVxJShruPKCol8pKeA3zw_4,7251
+rslearn/models/upsample.py,sha256=JvfnktT6Dgcql3cSoySWXZ7dmkDkfpRo6vDkpz8KFAQ,1326
+rslearn/models/use_croma.py,sha256=OSBqMuLp-pDtqPNWAVBfmX4wckmyYCKtUDdGCjJk_K8,17966
+rslearn/models/clay/clay.py,sha256=q2vcqyRFhCJvzLHwHLxHfWGyvVt819Xmk7TzuhXdPFI,8478
+rslearn/models/clay/configs/metadata.yaml,sha256=rZTFh4Yb9htEfbQNOPl4HTbFogEhzwIRqFzG-1uT01Y,4652
+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=ZiYNwJ3zWqZdvvzc0CXDzdvcwSCO_wSOCWlChemTsX8,19178
+rslearn/models/detr/matcher.py,sha256=4h_xFlgTMEJvJ6aLZUamrKZ72L5hDk9wPglNZ81JBg8,4533
+rslearn/models/detr/position_encoding.py,sha256=8FFoBT-Jtgqk7D4qDBTbVLOeAdmjdjtJTC608TaX6yY,3869
+rslearn/models/detr/transformer.py,sha256=aK4HO7AkCZn7xGHP3Iq91w2iFPVshugOILYAjVjroCw,13971
+rslearn/models/detr/util.py,sha256=NMHhHbkIo7PoBUVbDqa2ZknJBTswmaxFCGHrPtFKnGg,676
+rslearn/models/galileo/__init__.py,sha256=QQa0C29nuPRva0KtGiMHQ2ZB02n9SSwj_wqTKPz18NM,112
+rslearn/models/galileo/galileo.py,sha256=CTEuYGPmxhVv8YsZzFm0BeVW2yvdXp7j1GifT53EDLU,22525
+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=Mfq7mv438zhOHBsU2jT19NMslNlslWm1-sCE06wVlbo,17829
+rslearn/models/olmoearth_pretrain/norm.py,sha256=pV-u9eBGxfiSFDMEipDwyTgKGhN1G5kSoOC5CBE1m-0,3336
+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
+rslearn/models/panopticon_data/sensors/himawari.yaml,sha256=UzAgB1jpRQkuDb_MUOQcGfs7t8RO3JhPDHgA5WSJkc0,152
+rslearn/models/panopticon_data/sensors/intuition.yaml,sha256=Om82ICOxjy3uU5XAOUTgDsu_yx6Ms4cJUucbqHlToTA,8491
+rslearn/models/panopticon_data/sensors/landsat8.yaml,sha256=MB8lq9FgyaGivx3OobBLsbPm8T4jBopGlQXUmSnleeU,1394
+rslearn/models/panopticon_data/sensors/modis_terra.yaml,sha256=ltgje4i8qBabxvXI61JJdtsY-0AJ3zxcBSDfbALoSNA,1563
+rslearn/models/panopticon_data/sensors/qb2_ge1.yaml,sha256=5IzuBOXctayDW-UpH6vP5nDDJZP60ViG5Q9RuM2Ll78,613
+rslearn/models/panopticon_data/sensors/sentinel1.yaml,sha256=vbJuThuOo4JHb1cRajoOGb66_iuCVnUNDNR5GIg1YEs,1156
+rslearn/models/panopticon_data/sensors/sentinel2.yaml,sha256=qYJ92x-GHO0ZdCrTtCjG_47Yw4MiSFeoKjVbJV-_60Q,1670
+rslearn/models/panopticon_data/sensors/superdove.yaml,sha256=QpIRyopdV4hAez_EIsDwhGFT4VtTk7UgzQveyc8t8fc,795
+rslearn/models/panopticon_data/sensors/wv23.yaml,sha256=SWYSlkka6UViKAz6YI8aqwQ-Ayo-S5kmNa9rO3iGW6o,1172
+rslearn/models/presto/__init__.py,sha256=eZrB-XKi_vYqZhpyAOwppJi4dRuMtYVAdbq7KRygze0,64
+rslearn/models/presto/presto.py,sha256=fkyHB85Hfx5L-4yejSFAFv83gk9VFqAR1GTgggtq0EA,11049
+rslearn/models/presto/single_file_presto.py,sha256=-P00xjhj9dx3O6HqWpQmG9dPk_i6bT_t8vhX4uQm5tA,30242
+rslearn/tile_stores/__init__.py,sha256=-cW1J7So60SEP5ZLHCPdaFBV5CxvV3QlOhaFnUkhTJ0,1675
+rslearn/tile_stores/default.py,sha256=PYaDNvBxhJTDKJGw0EjDTSE1OKajR7_iJpMbOjj-mE8,15054
+rslearn/tile_stores/tile_store.py,sha256=9AeYduDYPp_Ia2NMlq6osptpz_AFGIOQcLJrqZ_m-z0,10469
+rslearn/train/__init__.py,sha256=fnJyY4aHs5zQqbDKSfXsJZXY_M9fbTsf7dRYaPwZr2M,30
+rslearn/train/all_patches_dataset.py,sha256=EVoYCmS3g4OfWPt5CZzwHVx9isbnWh5HIGA0RBqPFeA,21145
+rslearn/train/data_module.py,sha256=pgut8rEWHIieZ7RR8dUvhtlNqk0egEdznYF3tCvqdHg,23552
+rslearn/train/dataset.py,sha256=Jy1jU3GigfHaFeX9rbveX9bqy2Pd5Wh_vquD6_aFnS8,36522
+rslearn/train/lightning_module.py,sha256=HA9e-74oUZR5s7piQP9Mwxwz0vw0-p4HdmijfgBSpU0,14776
+rslearn/train/model_context.py,sha256=6o66BY6okBK-D5e0JUwPd7fxD_XehVaqxdQkJJKmQ3E,2580
+rslearn/train/optimizer.py,sha256=EKSqkmERalDA0bF32Gey7n6z69KLyaUWKlRsGJfKBmE,927
+rslearn/train/prediction_writer.py,sha256=rW0BUaYT_F1QqmpnQlbrLiLya1iBfC5Pb78G_NlF-vA,15956
+rslearn/train/scheduler.py,sha256=Yyq2fMHH08OyGt9qWD7iLc92XNIBbZHLtzlLR-f732s,2728
+rslearn/train/callbacks/__init__.py,sha256=VNV0ArZyYMvl3dGK2wl6F046khYJ1dEBlJS6G_SYNm0,47
+rslearn/train/callbacks/adapters.py,sha256=yfv8nyCj3jmo2_dNkFrjukKxh0MHsf2xKqWwMF0QUtY,1869
+rslearn/train/callbacks/freeze_unfreeze.py,sha256=8fIzBMhCKKjpTffIeAdhdSjsBd8NjTZZEPBQaSul6Zc,17418
+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=72ZBcbunMsdPYQN53S-4GfiLIDrr1X3Hni07dBJ0pu0,14261
+rslearn/train/tasks/detection.py,sha256=B0tfB7UGIbRtjnye3PhzLmfeQ4X7ImO3A-_LeNhBA54,21988
+rslearn/train/tasks/embedding.py,sha256=98ykdmfaxQjsH0UrUdTGmz1f0hCMPcNYTzt1YFqNQwQ,3869
+rslearn/train/tasks/multi_task.py,sha256=1ML9mZ-kM3JfElisLOWBUn4k12gsKTFjoYYgamnyxt8,6124
+rslearn/train/tasks/per_pixel_regression.py,sha256=znCLFaZbGx8lvIkntDXjcX7yy7giyyBdWN-TwTGaPV4,10197
+rslearn/train/tasks/regression.py,sha256=bVS_ApZSpbL0NaaM8Mu5Bsu4SBUyLpVtrPslulvvZHs,12695
+rslearn/train/tasks/segmentation.py,sha256=ie9ZV-sklLjQs35caiEglC1xff6dxeug_N-f_A8VosA,23034
+rslearn/train/tasks/task.py,sha256=B-z5XheXPTDaKauYzwze5ZD_O43R9uF76awgEk05bGE,3954
+rslearn/train/transforms/__init__.py,sha256=BkCAzm4f-8TEhPIuyvCj7eJGh36aMkZFYlq-H_jkSvY,778
+rslearn/train/transforms/concatenate.py,sha256=hVVBaxIdk1Cx8JHPirj54TGpbWAJx5y_xD7k1rmGmT0,3166
+rslearn/train/transforms/crop.py,sha256=d5mrC8k7g4zz9J0RJrjz_pS1q06HfhUdzH6dUlL3Wqc,4668
+rslearn/train/transforms/flip.py,sha256=KjMoZ4eFbu8119IlUDU4BBK06U76WjWEFmvqr9oiOMk,3836
+rslearn/train/transforms/mask.py,sha256=KMz-KeC4cuahA8CdRWXK9omeMfwheEM8avhj6yXGmm8,2407
+rslearn/train/transforms/normalize.py,sha256=H9hAYDVA__v0IpqC-7q_PyZ05qG6YTsORVQ7DTiLP6o,6382
+rslearn/train/transforms/pad.py,sha256=5Mks7Vp8_hRlsRjw9PERsQ7mVVKTAfg7Q6jUguXMhGY,5109
+rslearn/train/transforms/resize.py,sha256=AeMuhMlG-G7NRIMNucPESethohGe1ZKew-EVK2HzeSM,2636
+rslearn/train/transforms/select_bands.py,sha256=vqerph1UUS9i2G9xwyH6_5AoOuH9tLER8zppR4XiZs4,2802
+rslearn/train/transforms/sentinel1.py,sha256=PMMFo4RysMpZS7PBZ3yHGzJ5JADPZAOZpCbGhqL5vsc,2384
+rslearn/train/transforms/transform.py,sha256=n1Qzqix2dVvej-Q7iPzHeOQbqH79IBlvqPoymxhNVpE,4446
+rslearn/utils/__init__.py,sha256=GNvdTUmXakiEMnLdje7k1fe5aC7SFVqP757kbpN6Fzw,558
+rslearn/utils/array.py,sha256=RC7ygtPnQwU6Lb9kwORvNxatJcaJ76JPsykQvndAfes,2444
+rslearn/utils/feature.py,sha256=lsg0WThZDJzo1mrbaL04dXYI5G3x-n5FG9aEjj7uUaI,1649
+rslearn/utils/fsspec.py,sha256=h3fER_bkewzR9liEAULXguTIvXLUXA17pC_yZoWN5Tk,5902
+rslearn/utils/geometry.py,sha256=Z4alE4u0xa3fKGwoa4WQN5ZCjeqkPCJWfgcn8aw9t8I,21487
+rslearn/utils/get_utm_ups_crs.py,sha256=kUrcyjCK7KWvuP1XR-nURPeRqYeRO-3L8QUJ1QTF9Ps,3599
+rslearn/utils/grid_index.py,sha256=hRmrtgpqN1pLa-djnZtgSXqKJlbgGyttGnCEmPLD0zo,2347
+rslearn/utils/jsonargparse.py,sha256=TRyZA151KzhjJlZczIHYguEA-YxCDYaZ2IwCRgx76nM,4791
+rslearn/utils/mp.py,sha256=XYmVckI5TOQuCKc49NJyirDJyFgvb4AI-gGypG2j680,1399
+rslearn/utils/raster_format.py,sha256=qZpbODF4I7BsOxf43O6vTmH2TSNw6N8PP0wsFUVvdIw,26267
+rslearn/utils/rtree_index.py,sha256=j0Zwrq3pXuAJ-hKpiRFQ7VNtvO3fZYk-Em2uBPAqfx4,6460
+rslearn/utils/spatial_index.py,sha256=eomJAUgzmjir8j9HZnSgQoJHwN9H0wGTjmJkMkLLfsU,762
+rslearn/utils/sqlite_index.py,sha256=YGOJi66544e6JNtfSft6YIlHklFdSJO2duxQ4TJ2iu4,2920
+rslearn/utils/time.py,sha256=2ilSLG94_sxLP3y5RSV5L5CG8CoND_dbdzYEHVtN-I8,387
+rslearn/utils/vector_format.py,sha256=4ZDYpfBLLxguJkiIaavTagiQK2Sv4Rz9NumbHlq-3Lw,15041
+rslearn-0.0.21.dist-info/licenses/LICENSE,sha256=_99ZWPoLdlUbqZoSC5DF4ihiNwl5rTEmBaq2fACecdg,11352
+rslearn-0.0.21.dist-info/licenses/NOTICE,sha256=wLPr6rwV_jCg-xEknNGwhnkfRfuoOE9MZ-lru2yZyLI,5070
+rslearn-0.0.21.dist-info/METADATA,sha256=SUDvVod3HgoDdLYjRz9dwOL0c6V0SF-dTYuzsA4JJYk,37853
+rslearn-0.0.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rslearn-0.0.21.dist-info/entry_points.txt,sha256=doTBQ57NT7nq-dgYGgTTw6mafcGWb_4PWYtYR4rGm50,46
+rslearn-0.0.21.dist-info/top_level.txt,sha256=XDKo90WBH8P9RQumHxo0giLJsoufT4r9odv-WE6Ahk4,8
+rslearn-0.0.21.dist-info/RECORD,,

{rslearn-0.0.1.dist-info → rslearn-0.0.21.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.2.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

rslearn-0.0.21.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/data_sources/raster_source.py

@@ -1,309 +0,0 @@
-"""Helper functions for raster data sources."""
-
-from collections.abc import Callable
-from datetime import datetime
-from typing import Any
-
-import affine
-import numpy as np
-import numpy.typing as npt
-import rasterio.enums
-import rasterio.io
-import rasterio.transform
-from rasterio.crs import CRS
-
-from rslearn.config import BandSetConfig, RasterFormatConfig, RasterLayerConfig
-from rslearn.const import TILE_SIZE
-from rslearn.dataset import Window
-from rslearn.tile_stores import LayerMetadata, TileStore
-from rslearn.utils import Projection, STGeometry
-from rslearn.utils.raster_format import load_raster_format
-
-
-class ArrayWithTransform:
-    """Stores an array along with the transform associated with the array."""
-
-    def __init__(
-        self, array: npt.NDArray[Any], crs: CRS, transform: rasterio.transform.Affine
-    ) -> None:
-        """Create a new ArrayWithTransform instance.
-
-        Args:
-            array: the numpy array (C, H, W) storing the raster data.
-            crs: the CRS of the array
-            transform: the transform from pixel coordinates to projection coordinates.
-        """
-        self.array = array
-        self.crs = crs
-        self.transform = transform
-
-        # Store additional data matching rasterio.io.DatasetReader so we can use them
-        # interchangeably in ingest_raster.
-        self.width = self.array.shape[2]
-        self.height = self.array.shape[1]
-
-        # Left/top in projection units.
-        self.left = self.transform.c
-        self.top = self.transform.f
-
-        # Resolution in projection units per pixel.
-        self.x_resolution = self.transform.a
-        self.y_resolution = self.transform.e
-
-        # Right/bottom and bounds in projection units.
-        self.right = self.left + self.width * self.x_resolution
-        self.bottom = self.top + self.height * self.y_resolution
-        self.bounds = [
-            min(self.left, self.right),
-            min(self.top, self.bottom),
-            max(self.left, self.right),
-            max(self.top, self.bottom),
-        ]
-
-    def read(self) -> npt.NDArray[Any]:
-        """Reads the array.
-
-        This is to mimic the rasterio.DatasetReader API.
-
-        Returns:
-            the array
-        """
-        return self.array
-
-    def close(self):
-        """This is to mimic the rasterio.DatasetReader API.
-
-        The close function is a no-op.
-        """
-        pass
-
-    def pixel_bounds(self) -> tuple[int, int, int, int]:
-        """Returns the bounds of the array in global pixel coordinates.
-
-        The bounds is computed based on the stored transform.
-
-        The returned coordinates are (left, top, right, bottom).
-        """
-        start = (int(self.left / self.x_resolution), int(self.top / self.y_resolution))
-        end = (start[0] + self.array.shape[2], start[1] + self.array.shape[1])
-        return (start[0], start[1], end[0], end[1])
-
-    def get_tile(self, tile: tuple[int, int]) -> npt.NDArray[Any]:
-        """Returns portion of image corresponding to a tile.
-
-        Args:
-            tile: the tile to retrieve
-
-        Returns:
-            The portion of the image corresponding to the requested tile.
-        """
-        bounds = self.pixel_bounds()
-        x1 = tile[0] * TILE_SIZE - bounds[0]
-        y1 = tile[1] * TILE_SIZE - bounds[1]
-        x2 = x1 + TILE_SIZE
-        y2 = y1 + TILE_SIZE
-        # Need to pad output if x1/y1/x2/y2 are out of bounds.
-        # The padding is (before_y, after_y, before_x, after_x)
-        padding = [0, 0, 0, 0]
-        if x1 < 0:
-            padding[2] = -x1
-            x1 = 0
-        if y1 < 0:
-            padding[0] = -y1
-            y1 = 0
-        if x2 > self.array.shape[2]:
-            padding[3] = x2 - self.array.shape[2]
-            x2 = self.array.shape[2]
-        if y2 > self.array.shape[1]:
-            padding[1] = y2 - self.array.shape[1]
-            y2 = self.array.shape[1]
-        tile = self.array[:, y1:y2, x1:x2]
-        return np.pad(
-            tile, ((0, 0), (padding[0], padding[1]), (padding[2], padding[3]))
-        )
-
-
-def get_needed_projections(
-    tile_store: TileStore,
-    raster_bands: list[str],
-    band_sets: list[BandSetConfig],
-    geometries: list[STGeometry],
-) -> list[Projection]:
-    """Determines the projections of an item that are needed for a given raster file.
-
-    Projections that appear in geometries are skipped if a corresponding layer is
-    present in tile_store with metadata marked completed.
-
-    Args:
-        tile_store: TileStore prefixed with the item name and file name
-        raster_bands: list of bands contained in the raster file
-        band_sets: configured band sets
-        geometries: list of geometries for which the item is needed
-
-    Returns:
-        list of Projection objects for which the item has not been ingested yet
-    """
-    # Identify which band set configs are relevant to this raster.
-    raster_bands = set(raster_bands)
-    relevant_band_sets = []
-    for band_set in band_sets:
-        is_match = False
-        for band in band_set.bands:
-            if band not in raster_bands:
-                continue
-            is_match = True
-            break
-        if not is_match:
-            continue
-        relevant_band_sets.append(band_set)
-
-    all_projections = {geometry.projection for geometry in geometries}
-    needed_projections = []
-    for projection in all_projections:
-        for band_set in relevant_band_sets:
-            final_projection, _ = band_set.get_final_projection_and_bounds(
-                projection, None
-            )
-            ts_layer = tile_store.get_layer((str(final_projection),))
-            if ts_layer and ts_layer.get_metadata().properties.get("completed"):
-                continue
-            needed_projections.append(final_projection)
-    return needed_projections
-
-
-def ingest_raster(
-    tile_store: TileStore,
-    raster: rasterio.io.DatasetReader | ArrayWithTransform,
-    projection: Projection,
-    time_range: tuple[datetime, datetime] | None = None,
-    layer_config: RasterLayerConfig | None = None,
-    array_callback: Callable[[npt.NDArray[Any]], npt.NDArray[Any]] | None = None,
-) -> None:
-    """Ingests an in-memory rasterio dataset into the tile store.
-
-    Args:
-        tile_store: the tile store to ingest into, prefixed with the item name and
-            raster band names
-        raster: the rasterio raster
-        projection: the projection to save the raster as
-        time_range: time range of the raster
-        layer_config: the RasterLayerConfig of this layer
-        array_callback: callback function to apply on the array read from the raster
-    """
-    # Get layer in tile store to save under.
-    ts_layer = tile_store.create_layer(
-        (str(projection),), LayerMetadata(projection, time_range, {})
-    )
-    if ts_layer.get_metadata().properties.get("completed"):
-        return
-
-    # Warp each raster to this projection if needed.
-    array = raster.read()
-    if array_callback:
-        array = array_callback(array)
-
-    needs_warping = True
-    if isinstance(raster.transform, affine.Affine):
-        raster_projection = Projection(
-            raster.crs, raster.transform.a, raster.transform.e
-        )
-        needs_warping = raster_projection != projection
-
-    if not needs_warping:
-        # Include the top-left pixel index.
-        warped_array = ArrayWithTransform(array, raster.crs, raster.transform)
-
-    else:
-        # Compute the suggested target transform.
-        # rasterio negates the y resolution itself so here we have to negate it.
-        (dst_transform, dst_width, dst_height) = (
-            rasterio.warp.calculate_default_transform(
-                # Source info.
-                src_crs=raster.crs,
-                width=raster.width,
-                height=raster.height,
-                left=raster.bounds[0],
-                bottom=raster.bounds[1],
-                right=raster.bounds[2],
-                top=raster.bounds[3],
-                # Destination info.
-                dst_crs=projection.crs,
-                resolution=(projection.x_resolution, -projection.y_resolution),
-            )
-        )
-
-        resampling_method = rasterio.enums.Resampling.bilinear
-        if layer_config:
-            resampling_method = layer_config.resampling_method
-
-        # Re-project the raster to the destination crs, resolution, and transform.
-        dst_array = np.zeros((array.shape[0], dst_height, dst_width), dtype=array.dtype)
-        rasterio.warp.reproject(
-            source=array,
-            src_crs=raster.crs,
-            src_transform=raster.transform,
-            destination=dst_array,
-            dst_crs=projection.crs,
-            dst_transform=dst_transform,
-            resampling=resampling_method,
-        )
-        warped_array = ArrayWithTransform(dst_array, projection.crs, dst_transform)
-
-    ts_layer.write_raster(warped_array.pixel_bounds(), warped_array.array)
-    ts_layer.set_property("completed", True)
-
-
-def materialize_raster(
-    raster: rasterio.io.DatasetReader | ArrayWithTransform,
-    window: Window,
-    layer_name: str,
-    band_cfg: BandSetConfig,
-):
-    """Materialize a given raster for a window.
-
-    Currently it is only supported for materializing one band set.
-
-    Args:
-        raster: the raster data
-        window: the window to materialize
-        layer_name: the layer
-        band_cfg: the band configuration
-    """
-    window_projection, window_bounds = band_cfg.get_final_projection_and_bounds(
-        window.projection, window.bounds
-    )
-
-    # Re-project to just extract the window.
-    array = raster.read()
-    window_width = window_bounds[2] - window_bounds[0]
-    window_height = window_bounds[3] - window_bounds[1]
-    dst_transform = rasterio.transform.Affine(
-        window_projection.x_resolution,
-        0,
-        window_bounds[0] * window_projection.x_resolution,
-        0,
-        window_projection.y_resolution,
-        window_bounds[1] * window_projection.y_resolution,
-    )
-    dst_array = np.zeros(
-        (array.shape[0], window_height, window_width), dtype=array.dtype
-    )
-    rasterio.warp.reproject(
-        source=array,
-        src_crs=raster.crs,
-        src_transform=raster.transform,
-        destination=dst_array,
-        dst_crs=window_projection.crs,
-        dst_transform=dst_transform,
-        resampling=rasterio.enums.Resampling.bilinear,
-    )
-
-    # Write the array to layer directory.
-    layer_dir = window.path / "layers" / layer_name
-    out_dir = layer_dir / "_".join(band_cfg.bands)
-    out_dir.mkdir(parents=True, exist_ok=True)
-    raster_format = load_raster_format(
-        RasterFormatConfig(band_cfg.format["name"], band_cfg.format)
-    )
-    raster_format.encode_raster(out_dir, window_projection, window_bounds, dst_array)
-    (layer_dir / "completed").touch()

rslearn/models/registry.py

@@ -1,5 +0,0 @@
-"""Model registry."""
-
-from class_registry import ClassRegistry
-
-Models = ClassRegistry()