baacb20d15066935e4a23c09b1c1a6843331172f,research/object_detection/meta_architectures/center_net_meta_arch.py,,prediction_tensors_to_temporal_offsets,#Any#Any#Any#,338

Before Change

  
  _, _, width, _ = _get_shape(offset_predictions, 4)

  peak_spatial_indices = flattened_indices_from_row_col_indices(
      y_indices, x_indices, width)
  y_indices = _to_float32(y_indices)
  x_indices = _to_float32(x_indices)

  offsets_flat = _flatten_spatial_dimensions(offset_predictions)

After Change

  ], axis=1)

  new_offsets = tf.gather_nd(offset_predictions, combined_indices)
  offsets = tf.reshape(new_offsets, [batch_size, num_boxes, -1])

  return offsets

Instances