class_docstr: CenterNet network using Hourglass from https://arxiv.org/abs/1904.07850S.
class_docstr_tok: ["CenterNet","network","using","Hourglass","from","https",":","\/\/arxiv.org\/abs\/1904.07850S","."]
code: def nms(centers):
      centers_max = tf.nn.max_pool2d(centers, 3, 1, "SAME")
      centers_keep = tf.cast(
          tf.abs(centers_max - centers) < 1e-6, tf.float32
      )
      centers_nms = tf.multiply(centers_keep, centers)
      return centers_nms
  
imports: import collections
  import os
  import random
  import string
  import time
  from absl import logging
  import numpy
  import tensorflow
  from tensorflow_graphics.projects.points_to_3Dobjects.models import centernet_utils
  from tensorflow_graphics.projects.points_to_3Dobjects.networks import hourglass
  from tensorflow_graphics.projects.points_to_3Dobjects.transforms import transforms
  from tensorflow_graphics.projects.points_to_3Dobjects.utils import image
  from google3.pyglib import gfile
label_desc: Use image node_id 12 for calling the CenterNetVID obj's underlying member method code with example usage: obj.nms(centers) and returns: centers_nms