Method
__init__
(
self,
keys: KeysCollection,
spatial_dims: int = 2,
sigma: Sequence[float] | float | Sequence[torch.Tensor] | torch.Tensor = 0.0,
prob_threshold: float = 0.5,
box_size: int | Sequence[int] = 48,
allow_missing_keys: bool = False,
)
Source from the content-addressed store, hash-verified
| 749 | backend = ProbNMS.backend |
| 750 | |
| 751 | def __init__( |
| 752 | self, |
| 753 | keys: KeysCollection, |
| 754 | spatial_dims: int = 2, |
| 755 | sigma: Sequence[float] | float | Sequence[torch.Tensor] | torch.Tensor = 0.0, |
| 756 | prob_threshold: float = 0.5, |
| 757 | box_size: int | Sequence[int] = 48, |
| 758 | allow_missing_keys: bool = False, |
| 759 | ) -> None: |
| 760 | super().__init__(keys, allow_missing_keys) |
| 761 | self.prob_nms = ProbNMS( |
| 762 | spatial_dims=spatial_dims, sigma=sigma, prob_threshold=prob_threshold, box_size=box_size |
| 763 | ) |
| 764 | |
| 765 | def __call__(self, data: Mapping[Hashable, NdarrayOrTensor]): |
| 766 | d = dict(data) |
Callers
nothing calls this directly
Tested by
no test coverage detected