Skip to main content

array_to_tensor

Function array_to_tensor 

Source
pub fn array_to_tensor(image: &Array3<u8>) -> Array4<f32>
Expand description

Convert a raw HWC u8 array to a normalized NCHW tensor.

§Arguments

  • image - HWC array with shape (H, W, C) and u8 values.

§Returns

Array4 with shape (1, C, H, W) and values in [0, 1].