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 - 如果 imagedelta 不是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.       ]]]