code: def plot_gt_heatmaps(image, heatmaps, num_classes=6):
      figure, axis = plt.subplots(
          1, num_classes, figsize=(num_classes * 4, 4)
      )
      for class_id in range(num_classes):
          axis[class_id].imshow(
              np.concatenate(
                  [
                      image,
                      0.5
                      * np.ones([image.shape[0], image.shape[1], 1]),
                  ],
                  axis=-1,
              )
          )
          centers = heatmaps[:, :, class_id].numpy()
          axis[class_id].imshow(
              resize_heatmap(centers, color=[255, 0, 0])
          )
      return figure
  
imports: import os
  import pickle
  import re
  import matplotlib.pyplot
  import numpy
  import tensorflow
  from tensorflow_graphics.projects.points_to_3Dobjects.utils import tf_utils
  from google3.pyglib import gfile
label_desc: Use image node_id 12 for calling a global function with example usage: plot_gt_heatmaps(image, heatmaps, num_classes) and returns: figure