ndarray-ndimage 0.2.0

Multidimensional image processing for ArrayBase, the n-dimensional array data structure provided by ndarray.
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

use std::time::Instant;

use ndarray::{Array1, Array3, ShapeBuilder};

use ndarray_ndimage::{pad, PadMode};

fn main() {
    println!("C order");
    let data_c = (0..200 * 200 * 200)
        .map(|v| v as u32)
        .collect::<Array1<_>>()
        .into_shape((200, 200, 200))
        .unwrap();
    test(&data_c);

    println!("\nF order");
    let mut data_f = Array3::zeros((200, 200, 200).f());
    data_f.assign(&data_c);
    test(&data_f);
}

fn test(data: &Array3<u32>) {
    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Constant(0));
    let elapsed = now.elapsed();
    println!(
        "Constant 0  {}s {}ms  {}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded[(0, 0, 0)]
    );

    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Constant(1));
    let elapsed = now.elapsed();
    println!(
        "Constant 1  {}s {}ms  {}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded[(0, 0, 0)]
    );

    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Edge);
    let elapsed = now.elapsed();
    println!(
        "Edge        {}s {}ms  {:?}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded.dim()
    );

    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Maximum);
    let elapsed = now.elapsed();
    println!(
        "Maximum     {}s {}ms  {:?}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded.dim()
    );

    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Mean);
    let elapsed = now.elapsed();
    println!(
        "Mean        {}s {}ms  {:?}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded.dim()
    );

    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Median);
    let elapsed = now.elapsed();
    println!(
        "Median      {}s {}ms  {:?}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded.dim()
    );

    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Minimum);
    let elapsed = now.elapsed();
    println!(
        "Minimum     {}s {}ms  {:?}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded.dim()
    );

    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Reflect);
    let elapsed = now.elapsed();
    println!(
        "Reflect     {}s {}ms  {:?}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded.dim()
    );

    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Symmetric);
    let elapsed = now.elapsed();
    println!(
        "Symmetric   {}s {}ms  {:?}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded.dim()
    );

    let now = Instant::now();
    let padded = pad(&data, &[(1, 1)], PadMode::Wrap);
    let elapsed = now.elapsed();
    println!(
        "Wrap        {}s {}ms  {:?}",
        elapsed.as_secs(),
        elapsed.subsec_millis(),
        padded.dim()
    );
}