├── .DS_Store ├── .gitignore ├── .gitmodules ├── BAND_NAMES_v1_bgr.json ├── BAND_NAMES_v1_full.json ├── INSTALL.md ├── LICENSE ├── MODALITIES.py ├── NEW_DATASET.md ├── README.md ├── TRAINING.md ├── __init__.py ├── custom_loss.py ├── datasets_old.py ├── diagnostics.py ├── engine_finetune.py ├── engine_pretrain.py ├── env.yml ├── examples ├── ckpt_loading_example.ipynb ├── data_loader_example.ipynb └── finetune_torchhub.py ├── geobenchdataset.py ├── helpers.py ├── hubconf.py ├── main_finetune.py ├── main_pretrain.py ├── mmearth_dataset.py ├── models ├── __init__.py ├── convnextv2.py ├── convnextv2_sparse.py ├── convnextv2_unet.py ├── fcmae.py ├── norm_layers.py └── sparse_norm_layers.py ├── optim_factory.py ├── requirements.txt ├── slurm_scripts ├── slurm_fine_tune_all.sh ├── slurm_fine_tune_all_benchmark.sh ├── slurm_fine_tune_seg.sh ├── slurm_fine_tune_seg_all.sh ├── slurm_fine_tune_subsample.sh ├── slurm_fine_tune_tiny.sh └── slurm_pretrain.sh ├── subsample.py ├── tests ├── __init__.py ├── convert_test.py ├── dataset_test.py ├── finetune_test.py ├── pretrain_test.py └── stratified_test.py └── visualize_masking.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/.DS_Store -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/.gitmodules -------------------------------------------------------------------------------- /BAND_NAMES_v1_bgr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/BAND_NAMES_v1_bgr.json -------------------------------------------------------------------------------- /BAND_NAMES_v1_full.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/BAND_NAMES_v1_full.json -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/LICENSE -------------------------------------------------------------------------------- /MODALITIES.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/MODALITIES.py -------------------------------------------------------------------------------- /NEW_DATASET.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/NEW_DATASET.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/README.md -------------------------------------------------------------------------------- /TRAINING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/TRAINING.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /custom_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/custom_loss.py -------------------------------------------------------------------------------- /datasets_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/datasets_old.py -------------------------------------------------------------------------------- /diagnostics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/diagnostics.py -------------------------------------------------------------------------------- /engine_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/engine_finetune.py -------------------------------------------------------------------------------- /engine_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/engine_pretrain.py -------------------------------------------------------------------------------- /env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/env.yml -------------------------------------------------------------------------------- /examples/ckpt_loading_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/examples/ckpt_loading_example.ipynb -------------------------------------------------------------------------------- /examples/data_loader_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/examples/data_loader_example.ipynb -------------------------------------------------------------------------------- /examples/finetune_torchhub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/examples/finetune_torchhub.py -------------------------------------------------------------------------------- /geobenchdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/geobenchdataset.py -------------------------------------------------------------------------------- /helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/helpers.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/hubconf.py -------------------------------------------------------------------------------- /main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/main_finetune.py -------------------------------------------------------------------------------- /main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/main_pretrain.py -------------------------------------------------------------------------------- /mmearth_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/mmearth_dataset.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/convnextv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/models/convnextv2.py -------------------------------------------------------------------------------- /models/convnextv2_sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/models/convnextv2_sparse.py -------------------------------------------------------------------------------- /models/convnextv2_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/models/convnextv2_unet.py -------------------------------------------------------------------------------- /models/fcmae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/models/fcmae.py -------------------------------------------------------------------------------- /models/norm_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/models/norm_layers.py -------------------------------------------------------------------------------- /models/sparse_norm_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/models/sparse_norm_layers.py -------------------------------------------------------------------------------- /optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/optim_factory.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | ninja 2 | timm==0.9.7 3 | wandb 4 | tensorboardX 5 | kornia 6 | torchmetrics 7 | h5py 8 | -------------------------------------------------------------------------------- /slurm_scripts/slurm_fine_tune_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/slurm_scripts/slurm_fine_tune_all.sh -------------------------------------------------------------------------------- /slurm_scripts/slurm_fine_tune_all_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/slurm_scripts/slurm_fine_tune_all_benchmark.sh -------------------------------------------------------------------------------- /slurm_scripts/slurm_fine_tune_seg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/slurm_scripts/slurm_fine_tune_seg.sh -------------------------------------------------------------------------------- /slurm_scripts/slurm_fine_tune_seg_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/slurm_scripts/slurm_fine_tune_seg_all.sh -------------------------------------------------------------------------------- /slurm_scripts/slurm_fine_tune_subsample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/slurm_scripts/slurm_fine_tune_subsample.sh -------------------------------------------------------------------------------- /slurm_scripts/slurm_fine_tune_tiny.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/slurm_scripts/slurm_fine_tune_tiny.sh -------------------------------------------------------------------------------- /slurm_scripts/slurm_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/slurm_scripts/slurm_pretrain.sh -------------------------------------------------------------------------------- /subsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/subsample.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/convert_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/tests/convert_test.py -------------------------------------------------------------------------------- /tests/dataset_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/tests/dataset_test.py -------------------------------------------------------------------------------- /tests/finetune_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/tests/finetune_test.py -------------------------------------------------------------------------------- /tests/pretrain_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/tests/pretrain_test.py -------------------------------------------------------------------------------- /tests/stratified_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/tests/stratified_test.py -------------------------------------------------------------------------------- /visualize_masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vishalned/MMEarth-train/HEAD/visualize_masking.py --------------------------------------------------------------------------------