def __init__(self, saturation):
    super(SaturationJitterAug, self).__init__(saturation=saturation)
    self.saturation = saturation
    self.coef = nd.array([[[0.299, 0.587, 0.114]]])
def __init__(self, saturation):
    super(SaturationJitterAug, self).__init__(saturation=saturation)
    self.saturation = saturation
    self.coef = nd.array([[[0.299, 0.587, 0.114]]])