f15f59953d59ccca0fd3321d860a9768a3326941,research/object_detection/meta_architectures/center_net_meta_arch.py,CenterNetMetaArch,_rescore_instances,#CenterNetMetaArch#Any#Any#Any#,3006

Before Change

        continue
      class_id = kp_params.class_id
      keypoint_indices = kp_params.keypoint_indices
      num_keypoints = len(keypoint_indices)
      kpt_mask = tf.reduce_sum(
          tf.one_hot(keypoint_indices, depth=total_num_keypoints), axis=0)
      kpt_mask_tiled = tf.tile(kpt_mask[tf.newaxis, tf.newaxis, :],
                               multiples=[batch, max_detections, 1])

After Change

      visible_keypoints = tf.math.greater(keypoint_scores, 0.0)
      num_visible_keypoints = tf.reduce_sum(
          class_and_keypoint_mask_float *
          tf.cast(visible_keypoints, tf.float32), axis=-1)
      num_visible_keypoints = tf.math.maximum(num_visible_keypoints, 1.0)
      scores_for_class = (1./num_visible_keypoints) * (
          tf.reduce_sum(class_and_keypoint_mask_float *
                        scores[:, :, tf.newaxis] *
                        keypoint_scores, axis=-1))

Instances