(
self,
image_keys: KeysCollection,
box_keys: KeysCollection,
box_ref_image_keys: KeysCollection,
prob: float = 0.1,
min_zoom: Sequence[float] | float = 0.9,
max_zoom: Sequence[float] | float = 1.1,
mode: SequenceStr = InterpolateMode.AREA,
padding_mode: SequenceStr = NumpyPadMode.EDGE,
align_corners: Sequence[bool | None] | bool | None = None,
keep_size: bool = True,
allow_missing_keys: bool = False,
**kwargs: Any,
)
| 535 | backend = RandZoom.backend |
| 536 | |
| 537 | def __init__( |
| 538 | self, |
| 539 | image_keys: KeysCollection, |
| 540 | box_keys: KeysCollection, |
| 541 | box_ref_image_keys: KeysCollection, |
| 542 | prob: float = 0.1, |
| 543 | min_zoom: Sequence[float] | float = 0.9, |
| 544 | max_zoom: Sequence[float] | float = 1.1, |
| 545 | mode: SequenceStr = InterpolateMode.AREA, |
| 546 | padding_mode: SequenceStr = NumpyPadMode.EDGE, |
| 547 | align_corners: Sequence[bool | None] | bool | None = None, |
| 548 | keep_size: bool = True, |
| 549 | allow_missing_keys: bool = False, |
| 550 | **kwargs: Any, |
| 551 | ) -> None: |
| 552 | self.image_keys = ensure_tuple(image_keys) |
| 553 | self.box_keys = ensure_tuple(box_keys) |
| 554 | MapTransform.__init__(self, self.image_keys + self.box_keys, allow_missing_keys) |
| 555 | RandomizableTransform.__init__(self, prob) |
| 556 | self.box_ref_image_keys = ensure_tuple_rep(box_ref_image_keys, len(self.box_keys)) |
| 557 | |
| 558 | self.rand_zoom = RandZoom(prob=1.0, min_zoom=min_zoom, max_zoom=max_zoom, keep_size=keep_size, **kwargs) |
| 559 | self.mode = ensure_tuple_rep(mode, len(self.image_keys)) |
| 560 | self.padding_mode = ensure_tuple_rep(padding_mode, len(self.image_keys)) |
| 561 | self.align_corners = ensure_tuple_rep(align_corners, len(self.image_keys)) |
| 562 | self.keep_size = keep_size |
| 563 | |
| 564 | def set_random_state(self, seed: int | None = None, state: np.random.RandomState | None = None) -> RandZoomBoxd: |
| 565 | super().set_random_state(seed, state) |
nothing calls this directly
no test coverage detected