37 lines
1.1 KiB
Rust
37 lines
1.1 KiB
Rust
|
use std::path::Path;
|
||
|
|
|
||
|
|
use criterion::Criterion;
|
||
|
|
use criterion::{black_box, criterion_group, criterion_main};
|
||
|
|
use imsearch::search_index::{Database, FeatureGenerator};
|
||
|
|
|
||
|
|
pub fn bench_images(c: &mut Criterion) {
|
||
|
|
c.bench_function("indexing images", |b| {
|
||
|
|
b.iter(|| {
|
||
|
|
let files: Vec<std::path::PathBuf> = std::fs::read_dir("res/benchmark/")
|
||
|
|
.unwrap()
|
||
|
|
.map(|f| f.unwrap().path())
|
||
|
|
.collect();
|
||
|
|
|
||
|
|
let feats: Vec<FeatureGenerator> = vec![
|
||
|
|
imsearch::feature::luminance_distribution,
|
||
|
|
imsearch::feature::color_distribution,
|
||
|
|
imsearch::feature::average_luminance,
|
||
|
|
imsearch::feature::aspect_ratio,
|
||
|
|
];
|
||
|
|
|
||
|
|
let db = Database::new(&files, feats).unwrap();
|
||
|
|
|
||
|
|
black_box(
|
||
|
|
db.search(
|
||
|
|
Path::new("res/benchmark/bird.png"),
|
||
|
|
imsearch::feature::luminance_distribution,
|
||
|
|
)
|
||
|
|
.unwrap(),
|
||
|
|
);
|
||
|
|
})
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
criterion_group!(benches, bench_images);
|
||
|
|
criterion_main!(benches);
|