code: def tf_heatmap_to_lms(heatmap):
      hs = tf.argmax(tf.reduce_max(heatmap, 2), 1)
      ws = tf.argmax(tf.reduce_max(heatmap, 1), 1)
      lms = tf.transpose(
          tf.to_float(tf.stack([hs, ws])), perm=[1, 2, 0]
      )
      return lms
  
imports: import tensorflow
  from external.landmark_detector import networks
  from external.landmark_detector.flags import FLAGS
label_desc: Use image node_id 1 for calling a global function with example usage: tf_heatmap_to_lms(heatmap) and returns: lms