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