code: def augment(im, augment_str):
      """Augments image (used for training)."""
      if augment_str:
          im = Image.fromarray((im * 255).astype(np.uint8))
          im = make_augment(augment_str)(im)
          im = np.asarray(im).astype(np.float32) / 255
      return im
  
file_path: pycls/pycls/datasets/transforms.py
imports: from math import ceil, sqrt
  import cv2
  import numpy
  from PIL import Image
  from pycls.datasets.augment import make_augment
label_desc: Use image node_id 4 for calling a global function with example usage: augment(im, augment_str) and returns: im