code: def model_output_transform(output):
      return output["out"]
  
imports: import os
  import albumentations
  import cv2
  from albumentations.pytorch import ToTensorV2
  from dataflow import get_inference_dataloader, ignore_mask_boundaries
  from torchvision.models.segmentation import deeplabv3_resnet101
label_desc: Use image node_id 1 for calling a global function with example usage: model_output_transform(output) and returns: output