def __call__(self, data): self.randomize() d = dict(data) if not self._do_transform: return d rotator = Rotate(self.angle, self.spatial_axes, self.reshape, self.order, self.mode, self.cval, self.prefilter) for key in self.keys: d[key] = rotator(d[key]) return d
def __call__(self, data): self.randomize() d = dict(data) if not self._do_transform: return d rotator = Rotate( angle=self.x if d[self.keys[0]].ndim == 3 else (self.x, self.y, self.z), keep_size=self.keep_size, ) for idx, key in enumerate(self.keys): d[key] = rotator( d[key], mode=self.mode[idx], padding_mode=self.padding_mode[idx], align_corners=self.align_corners[idx], ) return d
def __init__( self, keys, angle, spatial_axes=(0, 1), reshape=True, order=1, mode="constant", cval=0, prefilter=True ): super().__init__(keys) self.rotator = Rotate( angle=angle, spatial_axes=spatial_axes, reshape=reshape, order=order, mode=mode, cval=cval, prefilter=prefilter, )
def __init__( self, keys: KeysCollection, angle: Union[Sequence[float], float], keep_size: bool = True, mode: GridSampleModeSequence = GridSampleMode.BILINEAR, padding_mode: GridSamplePadModeSequence = GridSamplePadMode.BORDER, align_corners: Union[Sequence[bool], bool] = False, ): super().__init__(keys) self.rotator = Rotate(angle=angle, keep_size=keep_size) self.mode = ensure_tuple_rep(mode, len(self.keys)) self.padding_mode = ensure_tuple_rep(padding_mode, len(self.keys)) self.align_corners = ensure_tuple_rep(align_corners, len(self.keys))
def __call__(self, data): self.randomize() d = dict(data) if not self._do_transform: return d rotator = Rotate(angle=self.angle, spatial_axes=self.spatial_axes, reshape=self.reshape) for idx, key in enumerate(self.keys): d[key] = rotator(d[key], order=self.order[idx], mode=self.mode[idx], cval=self.cval[idx], prefilter=self.prefilter[idx]) return d
def __init__( self, keys: KeysCollection, angle, keep_size: bool = True, interp_order: str = "bilinear", mode: str = "border", align_corners: bool = False, ): super().__init__(keys) self.rotator = Rotate(angle=angle, keep_size=keep_size, align_corners=align_corners) self.interp_order = ensure_tuple_rep(interp_order, len(self.keys)) self.mode = ensure_tuple_rep(mode, len(self.keys))
def __init__(self, keys, angle, spatial_axes=(0, 1), reshape=True, order=1, mode="constant", cval=0, prefilter=True): super().__init__(keys) self.rotator = Rotate(angle=angle, spatial_axes=spatial_axes, reshape=reshape) self.order = ensure_tuple_rep(order, len(self.keys)) self.mode = ensure_tuple_rep(mode, len(self.keys)) self.cval = ensure_tuple_rep(cval, len(self.keys)) self.prefilter = ensure_tuple_rep(prefilter, len(self.keys))
def __init__( self, keys, angle, spatial_axes=(0, 1), reshape=True, interp_order=InterpolationCode.LINEAR, mode="constant", cval=0, prefilter=True, ): super().__init__(keys) self.rotator = Rotate(angle=angle, spatial_axes=spatial_axes, reshape=reshape) self.interp_order = ensure_tuple_rep(interp_order, len(self.keys)) self.mode = ensure_tuple_rep(mode, len(self.keys)) self.cval = ensure_tuple_rep(cval, len(self.keys)) self.prefilter = ensure_tuple_rep(prefilter, len(self.keys))