cubic -> bilinear

microsoft · Apr 22, 2024 · 264313f · 264313f
1 parent b37f223
commit 264313f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/docs/tutorials/custom_raster_dataset.ipynb b/docs/tutorials/custom_raster_dataset.ipynb
@@ -338,7 +338,7 @@
     "\n",
     "### `resampling`\n",
     "\n",
-    "Defaults to cubic for float Tensors and nearest for int Tensors. Can be overridden for custom resampling algorithms.\n",
+    "Defaults to bilinear for float Tensors and nearest for int Tensors. Can be overridden for custom resampling algorithms.\n",
     "\n",
     "### `separate_files`\n",
     "\n",

diff --git a/tests/datasets/test_geo.py b/tests/datasets/test_geo.py
@@ -292,7 +292,7 @@ def test_resampling_float_dtype(self, dtype: torch.dtype) -> None:
         ds = CustomRasterDataset(dtype, paths)
         x = ds[ds.bounds]
         assert x["image"].dtype == dtype
-        assert ds.resampling == Resampling.cubic
+        assert ds.resampling == Resampling.bilinear
 
     @pytest.mark.parametrize("dtype", [torch.long, torch.bool])
     def test_resampling_int_dtype(self, dtype: torch.dtype) -> None:

diff --git a/torchgeo/datasets/geo.py b/torchgeo/datasets/geo.py
@@ -389,7 +389,7 @@ def dtype(self) -> torch.dtype:
     def resampling(self) -> Resampling:
         """Resampling algorithm used when reading input files.
 
-        Defaults to cubic for float dtypes and nearest for int dtypes.
+        Defaults to bilinear for float dtypes and nearest for int dtypes.
 
         Returns:
             The resampling method to use.
@@ -398,7 +398,7 @@ def resampling(self) -> Resampling:
         """
         # Based on torch.is_floating_point
         if self.dtype in [torch.float64, torch.float32, torch.float16, torch.bfloat16]:
-            return Resampling.cubic
+            return Resampling.bilinear
         else:
             return Resampling.nearest