Class WriteFileMappingd

monai/transforms/io/dictionary.py:323–345

Dictionary-based wrapper of :py:class:`monai.transforms.WriteFileMapping`. Args: keys: keys of the corresponding items to be transformed. See also: :py:class:`monai.transforms.compose.MapTransform` mapping_file_path: Path to the JSON file where the mapping

class WriteFileMappingd(MapTransform):
    """
    Dictionary-based wrapper of :py:class:`monai.transforms.WriteFileMapping`.

    Args:
          keys: keys of the corresponding items to be transformed.
              See also: :py:class:`monai.transforms.compose.MapTransform`
          mapping_file_path: Path to the JSON file where the mappings will be saved.
              Defaults to "mapping.json".
          allow_missing_keys: don't raise exception if key is missing.
    """

    def __init__(
        self, keys: KeysCollection, mapping_file_path: Path | str = "mapping.json", allow_missing_keys: bool = False
    ) -> None:
        super().__init__(keys, allow_missing_keys)
        self.mapping = WriteFileMapping(mapping_file_path)

    def __call__(self, data: Mapping[Hashable, NdarrayOrTensor]) -> dict[Hashable, NdarrayOrTensor]:
        d = dict(data)
        for key in self.key_iterator(d):
            d[key] = self.mapping(d[key])
        return d


LoadImageD = LoadImageDict = LoadImaged