Skip to content

Commit ca11ac6

Browse files
committed
[DLMED] remove RandCompose
Signed-off-by: Nic Ma <[email protected]>
1 parent bf31399 commit ca11ac6

File tree

4 files changed

+4
-210
lines changed

4 files changed

+4
-210
lines changed

docs/source/transforms.rst

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,6 @@ Generic Interfaces
3838
:members:
3939
:special-members: __call__
4040

41-
`RandCompose`
42-
^^^^^^^^^^^^^
43-
.. autoclass:: RandCompose
44-
:members:
45-
:special-members: __call__
46-
4741
`InvertibleTransform`
4842
^^^^^^^^^^^^^^^^^^^^^
4943
.. autoclass:: InvertibleTransform

monai/transforms/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
# limitations under the License.
1111

1212
from .adaptors import FunctionSignature, adaptor, apply_alias, to_kwargs
13-
from .compose import Compose, RandCompose
13+
from .compose import Compose
1414
from .croppad.array import (
1515
BorderPad,
1616
BoundingRect,

monai/transforms/compose.py

Lines changed: 3 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,9 @@
2727
Transform,
2828
apply_transform,
2929
)
30-
from monai.utils import MAX_SEED, ensure_tuple, ensure_tuple_rep, get_seed
30+
from monai.utils import MAX_SEED, ensure_tuple, get_seed
3131

32-
__all__ = ["Compose", "RandCompose"]
32+
__all__ = ["Compose"]
3333

3434

3535
class Compose(Randomizable, InvertibleTransform):
@@ -143,7 +143,7 @@ def flatten(self):
143143
"""
144144
new_transforms = []
145145
for t in self.transforms:
146-
if type(t) == Compose:
146+
if isinstance(t, Compose):
147147
new_transforms += t.flatten().transforms
148148
else:
149149
new_transforms.append(t)
@@ -172,34 +172,3 @@ def inverse(self, data):
172172
for t in reversed(invertible_transforms):
173173
data = apply_transform(t.inverse, data, self.map_items, self.unpack_items)
174174
return data
175-
176-
177-
class RandCompose(Compose):
178-
def __init__(
179-
self,
180-
prob: Union[Sequence[float], float],
181-
transforms: Optional[Union[Sequence[Callable], Callable]] = None,
182-
map_items: bool = True,
183-
unpack_items: bool = False,
184-
) -> None:
185-
super().__init__(transforms=transforms, map_items=map_items, unpack_items=unpack_items)
186-
self.prob = ensure_tuple_rep(prob, len(self.transforms))
187-
self.applied: List[Callable] = []
188-
189-
def flatten(self):
190-
raise NotImplementedError("flatten method not yet implemented for `RandCompose` class.")
191-
192-
def __len__(self):
193-
return len(self.transforms)
194-
195-
def __call__(self, input_):
196-
rands = self.R.rand(len(self))
197-
self.applied = []
198-
for _transform, r, p in zip(self.transforms, rands, self.prob):
199-
if r < min(max(p, 0.0), 1.0):
200-
input_ = apply_transform(_transform, input_, self.map_items, self.unpack_items)
201-
self.applied.append(_transform)
202-
return input_
203-
204-
def _get_applied_transforms(self):
205-
return self.applied

tests/test_rand_compose.py

Lines changed: 0 additions & 169 deletions
This file was deleted.

0 commit comments

Comments
 (0)