extra_args not needed

yichiac · Mar 15, 2024 · f690d8b · f690d8b
1 parent dc0cb89
commit f690d8b
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/torchgeo/datamodules/agrifieldnet.py b/torchgeo/datamodules/agrifieldnet.py
@@ -7,7 +7,6 @@
 
 import kornia.augmentation as K
 import torch
-from kornia.constants import DataKey, Resample
 
 from ..datasets import AgriFieldNet, random_bbox_assignment
 from ..samplers import GridGeoSampler, RandomBatchGeoSampler
@@ -55,9 +54,6 @@ def __init__(
             K.RandomVerticalFlip(p=0.5),
             K.RandomHorizontalFlip(p=0.5),
             data_keys=["image", "mask"],
-            extra_args={
-                DataKey.MASK: {"resample": Resample.NEAREST, "align_corners": None}
-            },
         )
 
     def setup(self, stage: str) -> None: