f4ca9b361adb382423c753f7f864a5dd0e454964,torchsample/transforms/image_transforms.py,Grayscale,__call__,#Grayscale#,43

Before Change

    def __call__(self, x, y=None):
        x_dst = x[0]*0.299 + x[1]*0.587 + x[2]*0.114
        x_gs = x_dst.repeat(self.channels,1,1)

        if y is not None:
            y_dst = y[0]*0.299 + y[1]*0.587 + y[2]*0.114
            y_gs= y_dst.repeat(self.channels,1,1)

After Change

            _input_dst = _input[0]*0.299 + _input[1]*0.587 + _input[2]*0.114
            _input_gs = _input_dst.repeat(self.channels,1,1)
            outputs.append(_input_gs)
        return outputs if idx > 1 else outputs[0]

class RandomGrayscale(object):
    
    Randomly grayscales some images with given probability,

Instances