(self, array)
| 3505 | self.sort_fn = sort_fn |
| 3506 | |
| 3507 | def randomize(self, array): |
| 3508 | if self.min_offset is None: |
| 3509 | min_offset = (0,) * len(self.patch_size) |
| 3510 | else: |
| 3511 | min_offset = ensure_tuple_rep(self.min_offset, len(self.patch_size)) |
| 3512 | if self.max_offset is None: |
| 3513 | max_offset = tuple(s % p for s, p in zip(array.shape[1:], self.patch_size)) |
| 3514 | else: |
| 3515 | max_offset = ensure_tuple_rep(self.max_offset, len(self.patch_size)) |
| 3516 | |
| 3517 | self.offset = tuple(self.R.randint(low=low, high=high + 1) for low, high in zip(min_offset, max_offset)) |
| 3518 | |
| 3519 | def filter_count(self, image_np: NdarrayOrTensor, locations: np.ndarray) -> tuple[NdarrayOrTensor, np.ndarray]: |
| 3520 | if self.sort_fn == GridPatchSort.RANDOM: |
no test coverage detected