f951f5c69d6015e9f5aaacd5eeafb0abf94f2001,chainercv/links/model/ssd/multibox_coder.py,MultiboxCoder,decode,#MultiboxCoder#,187

Before Change

        // softmax
        mb_score = xp.exp(mb_conf)
        mb_prob = mb_score / mb_score.sum(axis=1, keepdims=True)

        bbox = []
        label = []
        prob = []
        for l in range(mb_conf.shape[1] - 1):
            bbox_l = mb_bbox
            // the l-th class corresponds for the (l + 1)-th column.
            prob_l = mb_prob[:, l + 1]

            mask = prob_l >= score_thresh
            bbox_l = bbox_l[mask]
            prob_l = prob_l[mask]

            if nms_thresh is not None:
                indices = utils.non_maximum_suppression(
                    bbox_l, nms_thresh, prob_l)
                bbox_l = bbox_l[indices]
                prob_l = prob_l[indices]

            bbox.append(bbox_l)

After Change

        // softmax
        mb_score = xp.exp(mb_conf)
        mb_score /= mb_score.sum(axis=1, keepdims=True)

        bbox = []
        label = []
        score = []
        for l in range(mb_conf.shape[1] - 1):
            bbox_l = mb_bbox
            // the l-th class corresponds for the (l + 1)-th column.
            score_l = mb_score[:, l + 1]

            mask = score_l >= score_thresh
            bbox_l = bbox_l[mask]
            score_l = score_l[mask]

            if nms_thresh is not None:
                indices = utils.non_maximum_suppression(
                    bbox_l, nms_thresh, score_l)
                bbox_l = bbox_l[indices]
                score_l = score_l[indices]

            bbox.append(bbox_l)

Instances