def __init__(self, mean, std):
    super(ColorNormalizeAug, self).__init__(mean=mean, std=std)
    self.mean = (mean if ((mean is None) or isinstance(mean, nd.NDArray)) else nd.array(mean))
    self.std = (std if ((std is None) or isinstance(std, nd.NDArray)) else nd.array(std))
def __init__(self, mean, std):
    super(ColorNormalizeAug, self).__init__(mean=mean, std=std)
    self.mean = (mean if ((mean is None) or isinstance(mean, nd.NDArray)) else nd.array(mean))
    self.std = (std if ((std is None) or isinstance(std, nd.NDArray)) else nd.array(std))