Skip to main content

in_range

Function in_range 

Source
pub fn in_range(
    image: &Tensor,
    lower: &[f32],
    upper: &[f32],
) -> Result<Tensor, ImgProcError>
Expand description

Check if pixel values fall within a per-channel range. Create a binary mask where each pixel is 1.0 if all channels are within [lower, upper] (inclusive), 0.0 otherwise.

Input: [H, W, C], output: [H, W, 1].