ec5bcece006120b6cbb2a7b420e00ae218083b44,gluoncv/nn/coder.py,MultiPerClassDecoder,hybrid_forward,#MultiPerClassDecoder#Any#Any#,366

Before Change

        cls_ids = []
        for i in range(self._fg_class):
            cls_ids.append(template + i)  // b x N x 1
        cls_id = F.concat(*cls_ids, dim=-1)  // b x N x fg_class
        mask = scores > self._thresh
        cls_id = F.where(mask, cls_id, F.ones_like(cls_id) * -1)
        scores = F.where(mask, scores, F.zeros_like(scores))

After Change

    def hybrid_forward(self, F, x):
        scores = x.slice_axis(axis=self._axis, begin=1, end=None)  // b x N x fg_class
        template = F.zeros_like(x.slice_axis(axis=-1, begin=0, end=1))
        cls_id = F.broadcast_add(template,
                                 F.reshape(F.arange(self._fg_class), shape=(1, 1, self._fg_class)))
        mask = scores > self._thresh
        cls_id = F.where(mask, cls_id, F.ones_like(cls_id) * -1)
        scores = F.where(mask, scores, F.zeros_like(scores))
        return cls_id, scores

Instances