code: def color2d_setup():
      images = {}
  
      x = th.zeros(3, 30, 30)
      x[:, 10:21, 10:21] = 1
      images["color_01"] = x
  
      x = th.zeros(3, 30, 40)
      x[:, 10:21, 10:21] = 1
      images["color_02"] = x
  
      return images
  
file_path: pywick/tests/unit/transforms/test_image_transforms.py
imports: import torch
  from pywick.transforms import Grayscale, RandomGrayscale, Gamma, RandomGamma, RandomChoiceGamma, Brightness, RandomBrightness, RandomChoiceBrightness, Saturation, RandomSaturation, RandomChoiceSaturation, Contrast, RandomContrast, RandomChoiceContrast
label_desc: Use image node_id 3 for calling a global function with example usage: color2d_setup() and returns: images