(
self,
box_keys: KeysCollection,
box_mask_keys: KeysCollection,
label_keys: KeysCollection,
min_fg_label: int,
box_dtype: DtypeLike | torch.dtype = torch.float32,
label_dtype: DtypeLike | torch.dtype = torch.long,
allow_missing_keys: bool = False,
)
| 976 | """ |
| 977 | |
| 978 | def __init__( |
| 979 | self, |
| 980 | box_keys: KeysCollection, |
| 981 | box_mask_keys: KeysCollection, |
| 982 | label_keys: KeysCollection, |
| 983 | min_fg_label: int, |
| 984 | box_dtype: DtypeLike | torch.dtype = torch.float32, |
| 985 | label_dtype: DtypeLike | torch.dtype = torch.long, |
| 986 | allow_missing_keys: bool = False, |
| 987 | ) -> None: |
| 988 | super().__init__(box_keys, allow_missing_keys) |
| 989 | self.box_keys = ensure_tuple(box_keys) |
| 990 | self.label_keys = ensure_tuple(label_keys) |
| 991 | self.box_mask_keys = ensure_tuple(box_mask_keys) |
| 992 | if not len(self.label_keys) == len(self.box_keys) == len(self.box_mask_keys): |
| 993 | raise ValueError("Please make sure len(label_keys)==len(box_keys)==len(box_mask_keys)!") |
| 994 | self.bg_label = min_fg_label - 1 # make sure background label is always smaller than fg labels. |
| 995 | self.converter = MaskToBox(bg_label=self.bg_label, box_dtype=box_dtype, label_dtype=label_dtype) |
| 996 | self.box_dtype = box_dtype |
| 997 | |
| 998 | def __call__(self, data: Mapping[Hashable, NdarrayOrTensor]) -> dict[Hashable, NdarrayOrTensor]: |
| 999 | d = dict(data) |
nothing calls this directly
no test coverage detected