fa77503653547fbc02164c56b34f7c55cf35ee9f,chainercv/links/ssd/ssd.py,SSD,_suppress,#SSD#Any#Any#,112

Before Change

            bbox_label, score_label = raw_bbox[mask], raw_score[mask, i]

            if self.nms_threshold is not None:
                order = score_label.argsort()[::-1]
                bbox_label, score_label = bbox_label[order], score_label[order]
                bbox_label, param = transforms.non_maximum_suppression(
                    bbox_label, self.nms_threshold, return_param=True)
                score_label = score_label[param["selection"]]

After Change

            bbox_label, score_label = raw_bbox[mask], raw_score[mask][:, i]

            if self.nms_threshold is not None:
                order = xp.array(chainer.cuda.to_cpu(
                    score_label).argsort()[::-1])
                bbox_label, score_label = bbox_label[order], score_label[order]
                bbox_label, param = transforms.non_maximum_suppression(

Instances