Function tch_utils::color::hed_from_rgb

pub fn hed_from_rgb(rgb: &Tensor) -> Tensor

Convert a tensor of RGB values to HED values.

Arguments

• rgb: Tensor - The tensor of RGB values [N, 3, H, W] with float values in the range [0, 1]

Returns

Tensor - The tensor of HED values [N, 3, H, W] with float values in the range [0, 1]