mindspore.ops.AdjustHue
- class mindspore.ops.AdjustHue[源代码]
调整 RGB 图像的色调。
说明
该运算是种将 RGB 图像转换为浮点表示的便捷方法。通过将图像转换为HSV色彩空间并移动色调通道中的强度来调整图像,然后转换回原始数据模式。 当多个调整依次进行时尽量减少冗余转换的数量。
- 输入:
image (Tensor) - 输入的Tensor。shape的最后一个维度的必须为3。dtype需要是float16或float32。Tensor的维度至少是三维。
delta (Tensor) - 色调通道的添加值。dtype需要是float32。Tensor必须是零维的。
- 输出:
Tensor,具有与 image 相同的shape和dtype。
- 异常:
TypeError - 如果 image 或 delta 不是Tensor。
TypeError - 如果 image 的dtype不是:float32或float16。
TypeError - 如果 delta 的dtype不是:float32。
ValueError - 如果 image 的维度低于三维。
- 支持平台:
GPU
CPU
样例:
>>> class AdjustHue(nn.Cell): ... def __init__(self): ... super(AdjustHue, self).__init__() ... self.adjustHue = ops.AdjustHue() ... def construct(self, image, delta): ... return self.adjustHue(image, delta) ... >>> image = np.array([[[1, 2, 3], [4, 5, 6]], ... [[7, 8, 9], [10, 11, 12]], ... [[13, 14, 15], [16, 17, 18]]]).astype(np.float32) >>> delta = 0.2 >>> adjust_hue = AdjustHue() >>> output = adjust_hue(Tensor(image), Tensor(delta)) >>> print("output", output) output [[[ 2.3999996 1. 3. ] [ 5.3999996 4. 6. ]] [[ 8.4 7. 9. ] [11.4 10. 12. ]] [[14.4 13. 15. ] [17.4 16. 18. ]]]