From c887f5fef803e720052898b9561028dfd50e51db Mon Sep 17 00:00:00 2001 From: gennyble Date: Tue, 16 Jan 2024 15:06:29 -0600 Subject: ability to choose kmeans implementation --- squash/src/main.rs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'squash/src/main.rs') diff --git a/squash/src/main.rs b/squash/src/main.rs index a7cc66b..24d557f 100644 --- a/squash/src/main.rs +++ b/squash/src/main.rs @@ -31,10 +31,12 @@ fn main() -> Result<(), anyhow::Error> { builder = builder.selector(sorsel); } - cli::Selector::Kmeans => builder = builder.selector(Kmeans), + cli::Selector::Kmeans => builder = builder.selector(Kmeans { max_iter: 10 }), }; + let start = std::time::Instant::now(); let mut squasher = builder.build(&image.data); + println!("{:.2}ms", start.elapsed().as_secs_f32()); let size = squasher.map_over(&mut image.data); image.data.resize(size, 0); -- cgit 1.4.1-3-g733a5