From 33c27c2a88c4e84a1e5682a2e1cfaffb8f17d964 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sun, 25 Feb 2024 04:17:17 -0600 Subject: [PATCH 1/3] Clean up unnecessary code --- torchgeo/datamodules/nccm.py | 8 ++------ torchgeo/samplers/batch.py | 1 - torchgeo/trainers/segmentation.py | 1 - 3 files changed, 2 insertions(+), 8 deletions(-) diff --git a/torchgeo/datamodules/nccm.py b/torchgeo/datamodules/nccm.py index bea1b466a23..db4afdd0734 100644 --- a/torchgeo/datamodules/nccm.py +++ b/torchgeo/datamodules/nccm.py @@ -21,6 +21,7 @@ class NCCMSentinel2DataModule(GeoDataModule): """LightningDataModule implementation for the NCCM and Sentinel2 datasets. Uses the train/val/test splits from the dataset. + .. versionadded:: 0.6 """ @@ -67,12 +68,7 @@ def __init__( }, ) - self.val_aug = AugmentationSequential( - K.Normalize(mean=self.mean, std=self.std), - K.CenterCrop(self.patch_size), - data_keys=["image", "mask"], - ) - self.test_aug = AugmentationSequential( + self.aug = AugmentationSequential( K.Normalize(mean=self.mean, std=self.std), K.CenterCrop(self.patch_size), data_keys=["image", "mask"], diff --git a/torchgeo/samplers/batch.py b/torchgeo/samplers/batch.py index 8703999988b..ef9bb05f10f 100644 --- a/torchgeo/samplers/batch.py +++ b/torchgeo/samplers/batch.py @@ -122,7 +122,6 @@ def __init__( self.length += 1 self.hits.append(hit) areas.append(bounds.area) - if length is not None: self.length = length diff --git a/torchgeo/trainers/segmentation.py b/torchgeo/trainers/segmentation.py index 8a6ca077735..0be8a2e195d 100644 --- a/torchgeo/trainers/segmentation.py +++ b/torchgeo/trainers/segmentation.py @@ -254,7 +254,6 @@ def validation_step( batch["prediction"] = y_hat.argmax(dim=1) for key in ["image", "mask", "prediction"]: batch[key] = batch[key].cpu() - sample = unbind_samples(batch)[0] fig: Optional[Figure] = None From 4a488a3d6af0e0e245488b4d33453cb10c2fec45 Mon Sep 17 00:00:00 2001 From: shreyakannan1205 <111757918+shreyakannan1205@users.noreply.github.com> Date: Wed, 28 Feb 2024 10:08:17 -0600 Subject: [PATCH 2/3] Add files via upload From f83b00db1ef75e1bc0f697214ae60ae6c43dda98 Mon Sep 17 00:00:00 2001 From: shreyakannan1205 <111757918+shreyakannan1205@users.noreply.github.com> Date: Wed, 28 Feb 2024 10:09:03 -0600 Subject: [PATCH 3/3] new tif files