ndarray-image 0.1.0

Zero-copy conversion between ndarray and image crates
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use image::open;
use ndarray::s;
use ndarray_image::{ImgRgb, NdColor, NdImage};
use std::path::PathBuf;
use structopt::StructOpt;

#[derive(Debug, StructOpt)]
#[structopt(name = "image", about = "Loads image and puts red dots on it")]
struct Opt {
    /// File to put red dots on
    #[structopt(parse(from_os_str))]
    file: PathBuf,
    /// Output file with red dots
    #[structopt(parse(from_os_str))]
    output: PathBuf,
}

fn main() {
    let opt = Opt::from_args();
    let image = open(opt.file).expect("unable to open input image");
    let image = image.to_rgb();
    let ndimage: NdColor = NdImage(&image).into();
    let mut ndimage = ndimage.to_owned();
    let slice = ndimage.slice_mut(s![..;10, ..;2, 0]);
    for n in slice {
        *n = 255;
    }
    let image: Option<ImgRgb> = NdImage(ndimage.view()).into();
    let image = image.expect("failed to convert ndarray to image");
    image.save(opt.output).expect("failed to write output");
}