code: def draw_grid(images_aug, keypoints_aug):
      if keypoints_aug is None:
          keypoints_aug = []
          for bidx in range(len(images_aug)):
              keypoints_aug.append([None for image in images_aug[bidx]])
  
      images_kps_batches = []
      for bidx in range(len(images_aug)):
          images_kps_batch = []
          for image, kps in zip(images_aug[bidx], keypoints_aug[bidx]):
              if kps is None:
                  image_kps = image
              else:
                  image_kps = kps.draw_on_image(
                      image, size=5, color=[255, 0, 0]
                  )
              images_kps_batch.append(image_kps)
          images_kps_batches.extend(images_kps_batch)
  
      grid = ia.draw_grid(images_kps_batches, cols=len(images_aug[0]))
      return grid
  
file_path: imgaug/checks/check_multicore_pool.py
imports: from __future__ import print_function, division
  import time
  import multiprocessing
  import numpy
  from skimage import data
  import imgaug
  import imgaug.multicore
  from imgaug import augmenters
label_desc: Use image node_id 3 for calling a global function with example usage: draw_grid(images_aug, keypoints_aug) and returns: grid