364cd40384ad4f9cda4df949d0b1bd63abdcf6df,torchsample/transforms/image_transforms.py,Contrast,__call__,#Contrast#,216

Before Change

    def __call__(self, x, y=None):
        x_gs = Grayscale(keep_channels=True)(x)
        x_gs.fill_(x_gs[1].mean())

        alpha = 1.0 + random.uniform(-self.var, self.var)
        x = _blend(x, x_gs, alpha)

After Change

    def __call__(self, x, y=None):
        channel_means = x.mean(1)
        for i in range(2,x.dim()):
            channel_means = channel_means.mean(i)
        channel_means = channel_means.expand_as(x)

        x = (x - channel_means) * self.var + channel_means
        if y is not None:
            channel_means = x.mean(1).mean(2).expand_as(x)
            x = (x - channel_means) * self.var + channel_means          

Instances