77b8012989df60009c7a324de603c644f7aa86eb,luminoth/models/ssd/ssd_target.py,SSDTarget,_build,#SSDTarget#Any#Any#Any#Any#,31

Before Change

        // the greatest).

        // Transform to one-hot vector.
        cls_target_one_hot = tf.one_hot(
            tf.cast(best_proposals_gt_labels, tf.int32), depth=self._num_classes + 1,
            name="cls_target_one_hot"
        )

        // We get cross entropy loss of each proposal.
        cross_entropy_per_proposal = (
            tf.nn.softmax_cross_entropy_with_logits(
                labels=cls_target_one_hot, logits=probs
            )
        )

After Change

        // Use the worst backgrounds (the bgs which probability of being fg is
        // the greatest).
        cls_probs = probs[:, 1:]
        max_cls_probs = tf.reduce_max(cls_probs, axis=1)

        // Exclude boxes with IOU > `background_threshold_high` with any GT.

Instances