BAAI
/

repace nearest-exact by nearest

#4
Files changed (1) hide show
  1. model_segvol_single.py +1 -1
model_segvol_single.py CHANGED
@@ -147,7 +147,7 @@ class SegVolProcessor():
147
  transforms.ToTensord(keys=["image", "label"]),
148
  ]
149
  )
150
- self.zoom_out_transform = transforms.Resized(keys=["image", "label"], spatial_size=spatial_size, mode='nearest-exact')
151
  self.transform4train = transforms.Compose(
152
  [
153
  # transforms.AddChanneld(keys=["image"]),
 
147
  transforms.ToTensord(keys=["image", "label"]),
148
  ]
149
  )
150
+ self.zoom_out_transform = transforms.Resized(keys=["image", "label"], spatial_size=spatial_size, mode='nearest')
151
  self.transform4train = transforms.Compose(
152
  [
153
  # transforms.AddChanneld(keys=["image"]),