preprocess.py 文件源码

def _change_brightness(self, X):
        X = color.rgb2hsv(X.transpose(1, 2, 0)/255.0)
        diff = np.random.uniform(-1 * self._brightness, self._brightness)
        X[:,:,2] = np.clip((X[:,:,2] + diff), 0.0, 1.0)
        X = color.hsv2rgb(X).transpose(2, 0, 1) * 255.0

        return X.astype(np.int32)