r/visualhaar
1
0
Fork 0
Code Issues Pull requests Releases 1 Wiki Activity

cleanup test a bit

Browse source
This commit is contained in:
Ruben van de Ven 2020-10-01 16:21:36 +02:00
parent 928da82d24
commit 7c7ae29bf9
2 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/heatmap.rs
View file

@ -1,4 +1,3 @@
use ndarray as nd;
use image; use image;
pub enum ColorMaps{ pub enum ColorMaps{

4
src/test.rs
View file

@ -28,7 +28,7 @@ fn main() {
let mut sw = Stopwatch::start_new(); let mut sw = Stopwatch::start_new();
let frame = image::open("/home/ruben/Documents/Projecten/2020/rust/lena_orig-s.png"); let frame = image::open("test.png");
// let vec: Vec<u8> = frame.data().to_vec(); // let vec: Vec<u8> = frame.data().to_vec();
@ -48,7 +48,7 @@ fn main() {
let i = ib.as_rgb8().unwrap().clone(); let i = ib.as_rgb8().unwrap().clone();
let hm = Some(heatmap::Heatmap::new(heatmap::ColorMaps::Plasma)); let hm = Some(heatmap::Heatmap::new(heatmap::ColorMaps::Plasma));
let image = haar.scan_image(i, &hm).unwrap().dynamic_img; let image = haar.scan_image(i, &hm).unwrap().dynamic_img;
image.save("/home/ruben/Documents/Projecten/2020/rust/lena_orig-output.png"); image.save("test-output.png");
// let hm = heatmap::Heatmap::new(heatmap::ColorMaps::NipySpectral); // let hm = heatmap::Heatmap::new(heatmap::ColorMaps::NipySpectral);
// let hm = heatmap::Heatmap::new(heatmap::ColorMaps::TraficLight); // let hm = heatmap::Heatmap::new(heatmap::ColorMaps::TraficLight);