827e12f3b9506e0f78cc79445441cd9d1f25c055,chainercv/links/model/yolo/yolov3.py,YOLOv3,__call__,#YOLOv3#Any#,55

Before Change

        ys = []

        h = x
        hs = []
        for i, link in enumerate(self):
            h = link(h)
            if i in {35, 43, 51}:
                ys.append(h)

            if i in {14, 23, 33, 41}:
                hs.append(h)
            elif i in {35, 43}:
                h = hs.pop()
            elif i in {36, 44}:
                h = F.concat((_upsample(h), hs.pop()))

        return ys

After Change

    def __call__(self, x):
        ys = []
        for h, link in zip(self.extractor(x), self.subnet):
            h = link(h)
            h = F.reshape(h, (h.shape[0], 3, 1 + 4 + self.n_fg_class, -1))
            h = F.transpose(h, (0, 3, 1, 2))
            ys.append(h)
        return F.concat(ys)

Instances