9f0b230e1f59cb399e36210483bc075760b02eaf,gluonvision/model_zoo/ssd/ssd.py,SSD,hybrid_forward,#SSD#Any#Any#,154

Before Change

        cls_ids, scores = self.cls_decoder(F.softmax(cls_preds))
        result = F.concat(
            cls_ids.expand_dims(axis=-1), scores.expand_dims(axis=-1), bboxes, dim=-1)
        conf_mask = F.tile(scores.expand_dims(axis=-1) > 0.01, reps=(1, 1, 6))
        result = F.where(conf_mask, result, F.ones_like(result) * -1)
        if self.nms_thresh > 0 and self.nms_thresh < 1:
            result = F.contrib.box_nms(
                result, overlap_thresh=self.nms_thresh, topk=self.nms_topk,

After Change

                    per_result, overlap_thresh=self.nms_thresh, topk=self.nms_topk,
                    id_index=0, score_index=1, coord_start=2)
            results.append(per_result)
        result = F.concat(*results, dim=1)
        //
        // cls_ids, scores = self.cls_decoder(F.softmax(cls_preds))
        // result = F.concat(
        //     cls_ids.expand_dims(axis=-1), scores.expand_dims(axis=-1), bboxes, dim=-1)

Instances