(
self,
box_keys: KeysCollection,
box_mask_keys: KeysCollection,
label_keys: KeysCollection,
box_ref_image_keys: KeysCollection,
min_fg_label: int,
ellipse_mask: bool = False,
allow_missing_keys: bool = False,
)
| 894 | """ |
| 895 | |
| 896 | def __init__( |
| 897 | self, |
| 898 | box_keys: KeysCollection, |
| 899 | box_mask_keys: KeysCollection, |
| 900 | label_keys: KeysCollection, |
| 901 | box_ref_image_keys: KeysCollection, |
| 902 | min_fg_label: int, |
| 903 | ellipse_mask: bool = False, |
| 904 | allow_missing_keys: bool = False, |
| 905 | ) -> None: |
| 906 | super().__init__(box_keys, allow_missing_keys) |
| 907 | self.box_keys = ensure_tuple(box_keys) |
| 908 | self.label_keys = ensure_tuple(label_keys) |
| 909 | self.box_mask_keys = ensure_tuple(box_mask_keys) |
| 910 | if not len(self.label_keys) == len(self.box_keys) == len(self.box_mask_keys): |
| 911 | raise ValueError("Please make sure len(label_keys)==len(box_keys)==len(box_mask_keys)!") |
| 912 | self.box_ref_image_keys = ensure_tuple_rep(box_ref_image_keys, len(self.box_keys)) |
| 913 | self.bg_label = min_fg_label - 1 # make sure background label is always smaller than fg labels. |
| 914 | self.converter = BoxToMask(bg_label=self.bg_label, ellipse_mask=ellipse_mask) |
| 915 | |
| 916 | def __call__(self, data: Mapping[Hashable, NdarrayOrTensor]) -> dict[Hashable, NdarrayOrTensor]: |
| 917 | d = dict(data) |
nothing calls this directly
no test coverage detected