code: def per_batch_transform(self) -> Callable:
      return T.Compose([prepare_target, remove_extra_dimensions])
  
file_path: lightning-flash/src/flash/image/segmentation/input_transform.py
imports: from dataclasses import dataclass
  from typing import Any, Callable, Dict, Tuple
  from flash.core.data.io.input import DataKeys
  from flash.core.data.io.input_transform import InputTransform
  from flash.core.data.transforms import AlbumentationsAdapter, ApplyToKeys
  from flash.core.utilities.imports import _ALBUMENTATIONS_AVAILABLE, _TORCHVISION_AVAILABLE, requires
label_desc: Use image node_id 3 for calling the SemanticSegmentationInputTransform obj's underlying member method code with example usage: obj.per_batch_transform() and returns: T