from monai.transforms import (
PadListDataCollate,
RandRotate,
RandRotate90,
RandRotate90d,
RandRotated,
RandSpatialCrop,
RandSpatialCropd,
RandZoom,
RandZoomd,
)
from monai.utils import set_determinism
TESTS: List[Tuple] = []
for pad_collate in [pad_list_data_collate, PadListDataCollate()]:
TESTS.append((dict, pad_collate,
RandSpatialCropd("image", roi_size=[8, 7],
random_size=True)))
TESTS.append((dict, pad_collate,
RandRotated("image", prob=1, range_x=np.pi,
keep_size=False)))
TESTS.append((dict, pad_collate,
RandZoomd("image",
prob=1,
min_zoom=1.1,
max_zoom=2.0,
keep_size=False)))
TESTS.append((dict, pad_collate, RandRotate90d("image", prob=1, max_k=2)))
TESTS.append(
RandRotated,
RandSpatialCrop,
RandSpatialCropd,
RandZoom,
RandZoomd,
ToTensor,
ToTensord,
)
from monai.utils import set_determinism
TESTS: List[Tuple] = []
for pad_collate in [
lambda x: pad_list_data_collate(batch=x, method="end", mode="constant"
),
PadListDataCollate(method="end", mode="constant"),
]:
TESTS.append((dict, pad_collate,
RandSpatialCropd("image", roi_size=[8, 7],
random_size=True)))
TESTS.append((dict, pad_collate,
RandRotated("image", prob=1, range_x=np.pi,
keep_size=False)))
TESTS.append((dict, pad_collate,
RandZoomd("image",
prob=1,
min_zoom=1.1,
max_zoom=2.0,
keep_size=False)))
TESTS.append(
(dict, pad_collate,