template-matching

GPU加速的模板匹配Rust库。该库旨在作为 imageproc::template_matching 的更快的替代品。

安装

[dependencies]
template-matching = { version = "0.1.0", features = ["image"] }

使用方法

use template_matching::{find_extremes, match_template, MatchTemplateMethod, TemplateMatcher};

fn main() {
    // Load images and convert them to f32 grayscale
    let input_image = image::load_from_memory(include_bytes!("input.png")).unwrap().to_luma32f();
    let template_image = image::load_from_memory(include_bytes!("template.png")).unwrap().to_luma32f();

    let result = match_template(&input_image, &template_image, MatchTemplateMethod::SumOfSquaredDifferences);

    // Or alternatively you can create the matcher first
    let matcher = TemplateMatcher::new();
    let result = matcher.match_template(&input_image, &template_image, MatchTemplateMethod::SumOfSquaredDifferences);

    // Calculate min & max values
    let extremes = find_extremes(&result);
}