(self, img: NdarrayOrTensor, mask: NdarrayOrTensor | None = None)
| 2555 | self.dtype = dtype |
| 2556 | |
| 2557 | def __call__(self, img: NdarrayOrTensor, mask: NdarrayOrTensor | None = None) -> NdarrayOrTensor: |
| 2558 | img = convert_to_tensor(img, track_meta=get_track_meta()) |
| 2559 | img_np, *_ = convert_data_type(img, np.ndarray) |
| 2560 | mask = mask if mask is not None else self.mask |
| 2561 | mask_np: np.ndarray | None = None |
| 2562 | if mask is not None: |
| 2563 | mask_np, *_ = convert_data_type(mask, np.ndarray) |
| 2564 | |
| 2565 | ret = equalize_hist(img=img_np, mask=mask_np, num_bins=self.num_bins, min=self.min, max=self.max) |
| 2566 | out, *_ = convert_to_dst_type(src=ret, dst=img, dtype=self.dtype or img.dtype) |
| 2567 | |
| 2568 | return out |
| 2569 | |
| 2570 | |
| 2571 | class IntensityRemap(RandomizableTransform): |
nothing calls this directly
no test coverage detected