image-dwt 0.1.0

An implementation of the À Trous Discrete Wavelet Transform for images
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

use image::DynamicImage;

use image_dwt::kernels::LinearInterpolationKernel;
use image_dwt::recompose::ATrousRecompose;
use image_dwt::transform::ATrousTransform;

fn main() {
    let image = image::open("./Dis.jpg").unwrap();
    let transform = ATrousTransform::new(&image, 3, LinearInterpolationKernel);

    for layer in transform {
        let name = match layer.pixel_scale {
            Some(pixel_scale) => {
                format!("level{pixel_scale}")
            }
            None => "residue".to_string(),
        };

        let image = DynamicImage::ImageRgb32F(layer.image_buffer);
        image.to_rgb8().save(format!("{name}.jpg")).unwrap();
    }

    let recomposed = ATrousRecompose::new(&image::open("residue.jpg").unwrap())
        .recompose((0..3).map(|layer| image::open(format!("level{layer}.jpg")).unwrap()));

    recomposed.to_rgb8().save("recombined.jpg").unwrap()
}