fix the issue of the naming rule of key_meta_dict

Checking all dictionary keys that start with `{key}_` to support custom settings of `meta_keys` and `meta_key_postfix`.
This ensures that no matter how users configure the naming conventions in `LoadImaged`,
we can correctly synchronize metadata from the MetaTensor to the corresponding meta dictionary.

Signed-off-by: Wei_Chuan, Chiang <[email protected]>
Co-authored-by: einsyang723 <[email protected]>
Co-authored-by: IamTingTing  <[email protected]>

Author: 3 people

monai/transforms/spatial/dictionary.py

@@ -24,7 +24,6 @@
 import numpy as np
 import torch
 
-import monai.transforms as transforms
 from monai.config import DtypeLike, KeysCollection, SequenceStr
 from monai.config.type_definitions import NdarrayOrTensor
 from monai.data.box_utils import BoxMode, StandardMode
@@ -522,11 +521,13 @@ def __call__(self, data: Mapping[Hashable, torch.Tensor], lazy: bool | None = No
                 output_spatial_shape=output_shape_k if should_match else None,
                 lazy=lazy_,
             )
-            if isinstance(d[key], MetaTensor) and f"{key}_meta_dict" in d:
-                if "filename_or_obj" in d[key].meta and is_supported_format(
-                    d[key].meta["filename_or_obj"], ["nii", "nii.gz"]
-                ):
-                    d = transforms.sync_meta_info(key, d)
+            if isinstance(d[key], MetaTensor):
+                meta_keys = [k for k in d.keys() if k is not None and k.startswith(f"{key}_")]
+                for meta_key in meta_keys:
+                    if "filename_or_obj" in d[key].meta and is_supported_format(
+                        d[key].meta["filename_or_obj"], ["nii", "nii.gz"]
+                    ):
+                        d[meta_key].update(d[key].meta)
             if output_shape_k is None:
                 output_shape_k = d[key].peek_pending_shape() if isinstance(d[key], MetaTensor) else d[key].shape[1:]
         return d