code: def resize_heatmap(centers, color=(1, 0, 0), stride=4):
      assert len(centers.shape) == 2
      centers = np.repeat(
          np.repeat(centers, stride, axis=0), stride, axis=1
      )
      centers = np.expand_dims(centers, axis=-1)
      cmin, cmax = np.min(centers), np.max(centers)
      centers = np.concatenate(
          [
              np.ones(centers.shape) * color[0],
              np.ones(centers.shape) * color[1],
              np.ones(centers.shape) * color[2],
              (centers - cmin) / cmax,
          ],
          axis=-1,
      )
      return centers
  
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 2 for calling a global function with example usage: resize_heatmap(centers, color, stride) and returns: centers