d37d08d9b2a806b3345fba41711c0d517b92a65c,chainercv/links/model/vgg/vgg16.py,VGG16Layers,predict,#VGG16Layers#Any#,215

Before Change

            imgs = chainer.Variable(imgs)
            y = self(imgs).data

            if self.do_ten_crop:
                n = y.shape[0] // 10
                y_shape = y.shape[1:]
                y = y.reshape((n, 10) + y_shape)
                y = self.xp.sum(y, axis=1) / 10
        return cuda.to_cpu(y)


def _max_pooling_2d(x):

After Change

            imgs = chainer.Variable(imgs)
            activations = self(imgs)

        if isinstance(activations, dict):
            for name, activation in activations.items():
                activation = activation.data

                if self.do_ten_crop:
                    activation = self._gather_ten_crop(activation)
                activations[name] = cuda.to_cpu(activations)
        else:
            activations = cuda.to_cpu(activations.data)
            if self.do_ten_crop:
                activations = self._gather_ten_crop(activations)

        return activations

    def _gather_ten_crop(self, y):
        xp = chainer.cuda.get_array_module(y)

Instances