102 lines
2.7 KiB
Rust
102 lines
2.7 KiB
Rust
use image_similarity::store::DescriptorStore;
|
|
use image_similarity::descriptors::{Descriptor, DCT};
|
|
use log::info;
|
|
use std::path::PathBuf;
|
|
use std::thread;
|
|
use clap::Parser;
|
|
|
|
#[derive(Parser)]
|
|
#[command(version, about, long_about = None)]
|
|
struct Cfg {
|
|
/// Path that contains the input images. Will not traverse directories.
|
|
path: PathBuf
|
|
}
|
|
|
|
fn make_store<T: Descriptor>(descriptor: T, input_path: PathBuf, save_path: PathBuf) -> DescriptorStore {
|
|
let mut store =
|
|
DescriptorStore::new()
|
|
.with_file(save_path);
|
|
info!("{}", store);
|
|
store.insert_directory(input_path, descriptor);
|
|
info!("{}", store);
|
|
store.save().expect("Error saving");
|
|
store
|
|
}
|
|
|
|
struct StoreParams {
|
|
desc: DCT,
|
|
input_path: PathBuf,
|
|
save_path: PathBuf,
|
|
}
|
|
|
|
fn main() {
|
|
env_logger::init();
|
|
let cfg = Cfg::parse();
|
|
|
|
let mut threads = vec![];
|
|
|
|
let params = vec![
|
|
StoreParams {
|
|
desc: DCT::new().with_quality(50),
|
|
input_path: cfg.path.clone(),
|
|
save_path: "dct50.messagepack".into(),
|
|
},
|
|
// StoreParams {
|
|
// desc: DCT::new().with_quality(30),
|
|
// input_path: cfg.path.clone(),
|
|
// save_path: "dct30.messagepack".into(),
|
|
// },
|
|
// StoreParams {
|
|
// desc: DCT::new().with_quality(10),
|
|
// input_path: cfg.path.clone(),
|
|
// save_path: "dct10.messagepack".into(),
|
|
// },
|
|
];
|
|
|
|
// let dct_50 =
|
|
// DCT::new()
|
|
// .with_quality(50);
|
|
// let dct_30 =
|
|
// DCT::new()
|
|
// .with_quality(30);
|
|
// let dct_10 =
|
|
// DCT::new()
|
|
// .with_quality(10);
|
|
|
|
//let (tx, rx) = mpsc::channel::<DescriptorStore>();
|
|
|
|
for param in params {
|
|
threads.push(
|
|
thread::spawn(move || {
|
|
let store = make_store(param.desc, param.input_path, param.save_path);
|
|
store.print_most_dups();
|
|
//tx.send(store).unwrap();
|
|
})
|
|
)
|
|
}
|
|
|
|
// let path_50 = cfg.path.clone();
|
|
// threads.push(
|
|
// thread::spawn(move || {
|
|
// let store = make_store(dct_50, path_50, "dct50.messagepack".into());
|
|
// tx.send(store).expect("Thread send result error");
|
|
// })
|
|
// );
|
|
// let path_30 = cfg.path.clone();
|
|
// threads.push(
|
|
// thread::spawn(move || {
|
|
// dct_30_store = make_store(dct_30, path_30, "dct30.messagepack".into());
|
|
// })
|
|
// );
|
|
// let path_10 = cfg.path.clone();
|
|
// threads.push(
|
|
// thread::spawn(move || {
|
|
// dct_50_store = make_store(dct_10, path_10, "dct10.messagepack".into());
|
|
// })
|
|
// );
|
|
for thread in threads {
|
|
let _ = thread.join();
|
|
}
|
|
|
|
}
|