repace nearest-exact by nearest
#4
by
yuxindu
- opened
- 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
|
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"]),
|