1dffde32fc1d9bce8bfadb5c3f58160d26f70ea8,official/vision/detection/ops/postprocess_ops.py,,_generate_detections,#Any#Any#Any#Any#Any#Any#,46

Before Change

      boxes_i = boxes[:, :, min(num_classes_for_box - 1, i), :]
      scores_i = scores[:, :, i]
      // Obtains pre_nms_num_boxes before running NMS.
      scores_i, indices = tf.nn.top_k(
          scores_i,
          k=tf.minimum(tf.shape(input=scores_i)[-1], pre_nms_num_boxes))
      boxes_i = tf.gather(boxes_i, indices, batch_dims=1, axis=1)

      // Filter out scores.
      boxes_i, scores_i = box_utils.filter_boxes_by_scores(

After Change

      boxes_i = boxes[:, :, min(num_classes_for_box - 1, i), :]
      scores_i = scores[:, :, i]
      // Obtains pre_nms_num_boxes before running NMS.
      boxes_i = tf.gather(boxes_i, indices[:, :, i], batch_dims=1, axis=1)

      // Filter out scores.
      boxes_i, scores_i = box_utils.filter_boxes_by_scores(

Instances