Ensuring the cropping transform is invertible (#1262)

This is a small change to ensure the transform is invertible and works
with the latest monai code-base.

---------

Signed-off-by: myron <amyronenko@nvidia.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: myron

auto3dseg/tasks/hecktor22/hecktor_crop_neck_region.py

@@ -50,7 +50,9 @@ def __call__(self, data):
 
         box_start, box_end = self.extract_roi(im_pet=im_pet, box_size=box_size)
 
-        if "label" in d and "label" in self.keys:
+        use_label = "label" in d and "label" in self.keys and (d["image"].shape[1:] == d["label"].shape[1:])
+
+        if use_label:
             # if label mask is available, let's check if the cropped region includes all foreground
             before_sum = d["label"].sum().item()
             after_sum = (
@@ -83,7 +85,8 @@ def __call__(self, data):
         d[self.end_coord_key] = box_end
 
         for key, m in self.key_iterator(d, self.mode):
-            self.push_transform(d, key, extra_info={"box_start": box_start, "box_end": box_end})
+            if key == "label" and not use_label:
+                continue
             d[key] = self.cropper.crop_pad(img=d[key], box_start=box_start, box_end=box_end, mode=m)
 
         return d