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 _get_width_height(width_height, topk_inds):
      b, _, _, n = width_height.shape
      width_height = tf.reshape(width_height, [b, -1, n])
      width_height = tf.gather(width_height, topk_inds, batch_dims=1)
      return width_height
  
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 15 for calling the CenterNetVID obj's underlying member method code with example usage: obj._get_width_height(width_height, topk_inds) and returns: width_height