image_to_ndarray.rs

use image::{DynamicImage, GenericImageView};
use ndarray::{Array, Array3, Axis};

fn image_to_ndarray(img: &DynamicImage) -> Array3<u8> {
    let (width, height) = img.dimensions();
    let mut data = vec![0u8; (width * height * 3) as usize];
    img.to_rgb8().into_raw().iter().enumerate().for_each(|(i, pixel)| {
        data[i] = *pixel;
    });
    Array::from_shape_vec((height as usize, width as usize, 3), data).unwrap().reversed_axes()
}