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)
评论列表
文章目录