o  i@slddlmZddlZddlmZmZddlZddlZddl m Z ddl m Z ddl mZGddde ZdS) ) annotationsN)HashableMapping)KeysCollection) MapTransform) ImageMetaKeycs8eZdZdZ     ddfdd ZdddZZS)EnsureSameShapeda Checks if segmentation label images (in keys) have the same spatial shape as the main image (in source_key), and raise an error if the shapes are significantly different. If the shapes are only slightly different (within an allowed_shape_difference in each dim), then resize the label using nearest interpolation. This transform is designed to correct datasets with slight label shape mismatches. Generally image and segmentation label must have the same spatial shape, however some public datasets are having slight shape mismatches, which will cause potential crashes when calculating loss or metric functions. labelFimageTkeysrallow_missing_keysbool source_keystrallowed_shape_differenceintwarnreturnNonecs&tj||d||_||_||_dS)a Args: keys: keys of the corresponding items to be compared to the source_key item shape. allow_missing_keys: do not raise exception if key is missing. source_key: key of the item with the reference shape. allowed_shape_difference: raises error if shapes are different more than this value in any dimension, otherwise corrects for the shape mismatch using nearest interpolation. warn: if `True` prints a warning if the label image is resized )r r N)super__init__rrr)selfr r rrr __class__a/home/dell461/cl/sdc2/last_ska_mid/HISourceFinder-master-l/src/monai/apps/auto3dseg/transforms.pyr#s zEnsureSameShaped.__init__dataMapping[Hashable, torch.Tensor]dict[Hashable, torch.Tensor]c Cst|}||jjdd}||D]q}||jdd}||krd}t||drs