From 90c362de7fa0187d309cee4b1ddb5018d7b68244 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Fri, 24 Jan 2025 20:15:50 +0100 Subject: [PATCH 1/2] ETCI2021: fix file list when 'vv' in directory name --- torchgeo/datasets/etci2021.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/torchgeo/datasets/etci2021.py b/torchgeo/datasets/etci2021.py index 7855c8bb3cf..f57bbb28951 100644 --- a/torchgeo/datasets/etci2021.py +++ b/torchgeo/datasets/etci2021.py @@ -170,17 +170,15 @@ def _load_files(self, root: Path, split: str) -> list[dict[str, str]]: folders = [os.path.join(folder, 'tiles') for folder in folders] for folder in folders: vvs = sorted(glob.glob(os.path.join(folder, 'vv', '*.png'))) - vhs = [vv.replace('vv', 'vh') for vv in vvs] - water_masks = [ - vv.replace('_vv.png', '.png').replace('vv', 'water_body_label') - for vv in vvs - ] + vhs = sorted(glob.glob(os.path.join(folder, 'vh', '*.png'))) + water_masks = sorted( + glob.glob(os.path.join(folder, 'water_body_label', '*.png')) + ) if split != 'test': - flood_masks = [ - vv.replace('_vv.png', '.png').replace('vv', 'flood_label') - for vv in vvs - ] + flood_masks = sorted( + glob.glob(os.path.join(folder, 'flood_masks', '*.png')) + ) for vv, vh, flood_mask, water_mask in zip( vvs, vhs, flood_masks, water_masks From 90cb6456b0c4ee8466c47923032ee1bd0ce63a02 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Fri, 24 Jan 2025 20:29:25 +0100 Subject: [PATCH 2/2] Typo --- torchgeo/datasets/etci2021.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchgeo/datasets/etci2021.py b/torchgeo/datasets/etci2021.py index f57bbb28951..6e824369ddb 100644 --- a/torchgeo/datasets/etci2021.py +++ b/torchgeo/datasets/etci2021.py @@ -177,7 +177,7 @@ def _load_files(self, root: Path, split: str) -> list[dict[str, str]]: if split != 'test': flood_masks = sorted( - glob.glob(os.path.join(folder, 'flood_masks', '*.png')) + glob.glob(os.path.join(folder, 'flood_label', '*.png')) ) for vv, vh, flood_mask, water_mask in zip(