From 8efa5b58bc268ae538043a95ef4314a100763634 Mon Sep 17 00:00:00 2001
From: leonard <l.hackel@tu-berlin.de>
Date: Tue, 2 Jul 2024 16:57:02 +0200
Subject: [PATCH] path updating
---
scripts/train_BigEarthNetv2_0.py | 2 +-
scripts/utils.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/scripts/train_BigEarthNetv2_0.py b/scripts/train_BigEarthNetv2_0.py
index eb4bc4a..0dbc0a3 100644
--- a/scripts/train_BigEarthNetv2_0.py
+++ b/scripts/train_BigEarthNetv2_0.py
@@ -85,7 +85,7 @@ def main(
trainer = default_trainer(hparams, use_wandb, test_run)
hostname, data_dirs = get_benv2_dir_dict()
- data_dirs = resolve_data_dir(data_dirs, allow_mock=True)
+ data_dirs = resolve_data_dir(data_dirs, allow_mock=False)
dm = default_dm(hparams, data_dirs, img_size)
trainer.fit(model, dm)
diff --git a/scripts/utils.py b/scripts/utils.py
index 8560795..3fb1d8b 100644
--- a/scripts/utils.py
+++ b/scripts/utils.py
@@ -35,7 +35,7 @@ BENv2_DIR_DICT_PLUTO = {
BENv2_DIR_DEFAULT = Path("~/data/BigEarthNet-V2").expanduser()
BENv2_DIR_DICT_DEFAULT = {
- "images_lmdb": BENv2_DIR_DEFAULT / "BigEarthNet-V2-LMDB",
+ "images_lmdb": BENv2_DIR_DEFAULT / "BENv2.lmdb",
"metadata_parquet": BENv2_DIR_DEFAULT / "metadata.parquet",
"metadata_snow_cloud_parquet": BENv2_DIR_DEFAULT / "metadata_for_patches_with_snow_cloud_or_shadow.parquet",
}
--
GitLab