pytorch RGB图像转HSV

可以使用 `torchvision` 库中的 `transforms` 模块来实现 RGB 图像转换为 HSV 彩色空间的操作。具体步骤如下：

1. 导入必要的库

import torchvision.transforms.functional as TF
from PIL import Image
import numpy as np

2. 定义 RGB 转 HSV 的函数

def rgb2hsv(rgb_img):
    # 将 RGB 图像转换为 numpy 数组
    rgb_np = np.array(rgb_img)
    # 归一化到 [0, 1] 范围
    rgb_np = rgb_np / 255.0
    # 分离颜色通道
    r, g, b = rgb_np[..., 0], rgb_np[..., 1], rgb_np[..., 2]
    # 计算 V 值
    v = np.maximum(np.maximum(r, g), b)
    # 计算 S 值
    s = np.where(v == 0, 0, (v - np.minimum(np.minimum(r, g), b)) / v)
    # 计算 H 值
    h = np.zeros_like(v)
    mask = v == r
    h[mask] = (g[mask] - b[mask]) / (v[mask] - np.minimum(np.minimum(g[mask], b[mask]), axis=-1))
    mask = v == g
    h[mask] = 2.0 + (b[mask] - r[mask]) / (v[mask] - np.minimum(np.minimum(b[mask], r[mask]), axis=-1))
    mask = v == b
    h[mask] = 4.0 + (r[mask] - g[mask]) / (v[mask] - np.minimum(np.minimum(r[mask], g[mask]), axis=-1))
    h = np.mod((h / 6.0), 1.0)
    # 将 HSV 数组转换为 PIL.Image 格式
    hsv_np = np.stack([h, s, v], axis=-1)
    hsv_np = hsv_np * 255.0
    hsv_np = hsv_np.astype(np.uint8)
    hsv_img = Image.fromarray(hsv_np, mode='HSV')
    return hsv_img

3. 加载 RGB 图像并转换为 HSV 彩色空间

# 加载 RGB 图像
rgb_img = Image.open('example.jpg').convert('RGB')
# 转换为 HSV 彩色空间
hsv_img = rgb2hsv(rgb_img)

4. 可以将转换后的 HSV 图像保存到本地

hsv_img.save('example_hsv.jpg')