code: def normalize_image_for_view(image):
      """Normalizes dataset image into the format for viewing."""
      image *= np.reshape(STDDEV_RGB, (3, 1, 1))
      image += np.reshape(MEAN_RGB, (3, 1, 1))
      image = np.transpose(image, (1, 2, 0))
      return image.clip(0, 255).round().astype("uint8")
  
file_path: objax/examples/image_classification/imagenet_resnet50_data.py
imports: import enum
  from typing import Optional, Sequence, Tuple
  import jax
  import numpy
  import tensorflow.compat.v2
  import tensorflow_datasets
label_desc: Use image node_id 4 for calling a global function with example usage: normalize_image_for_view(image) and returns: image