| 584 | """ |
| 585 | |
| 586 | def __init__( |
| 587 | self, |
| 588 | roi_size: Sequence[int] | int, |
| 589 | max_roi_size: Sequence[int] | int | None = None, |
| 590 | random_center: bool = True, |
| 591 | random_size: bool = False, |
| 592 | lazy: bool = False, |
| 593 | ) -> None: |
| 594 | super().__init__(lazy) |
| 595 | self.roi_size = roi_size |
| 596 | self.max_roi_size = max_roi_size |
| 597 | self.random_center = random_center |
| 598 | self.random_size = random_size |
| 599 | self._size: Sequence[int] | None = None |
| 600 | self._slices: tuple[slice, ...] |
| 601 | |
| 602 | def randomize(self, img_size: Sequence[int]) -> None: |
| 603 | self._size = fall_back_tuple(self.roi_size, img_size) |