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

Before Change

        x = _blend(x, x_gs, alpha)

        if y is not None:
            y_gs = Grayscale(keep_channels=True)(y)
            y_gs.fill_(y_gs[1].mean())
            y = _blend(y, y_gs, alpha)
            return x, y
        return x

After Change

        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          


class Gamma(object):

Instances