add refresh_samples function

microsoft · Sep 18, 2024 · 60539ad · 60539ad
1 parent 28e61e4
commit 60539ad
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/torchgeo/samplers/single.py b/torchgeo/samplers/single.py
@@ -269,6 +269,14 @@ def __init__(
 
         self.chips = self.get_chips()
 
+    def refresh_samples(self) -> None:
+        """Refresh the samples in the sampler.
+
+        This method is useful when you want to refresh the random samples in the sampler
+        without creating a new sampler instance.
+        """
+        self.chips = self.get_chips()
+
     def get_chips(self) -> GeoDataFrame:
         """Generate chips from the dataset.