detnet.py 文件源码

def grid_2d(in_size, out_size=None):
  grid_ys, grid_xs = tf.meshgrid(tf.range(0, in_size[0]),
                                 tf.range(0, in_size[1]),
                                 indexing='ij')

  if not out_size is None:
    grid_yxs = tf.image.resize_images(tf.pack([grid_ys, grid_xs], axis=2),
                                      out_size[0], out_size[1])
    grid_ys, grid_xs = grid_yxs[:,:,0], grid_yxs[:,:,1]

  grid_ys = grid_ys / tf.to_float(in_size[0])
  grid_xs = grid_xs / tf.to_float(in_size[1])
  return grid_ys, grid_xs