如何在 ArrayFire match_template 之后获得像素坐标 X，Y?

Question

>我正在尝试使用 ArrayFire 库中的 matching_template 函数，但我不知道如何找到最佳匹配值的 X 和 Y 坐标。 我正在使用 imageproc 库来执行此功能，并且它具有将坐标返回给我的 find_extremes 函数。你会如何使用ArrayFire库做同样的事情？

我使用图像处理器的示例

let template = image::open("connect.png").unwrap().to_luma8();
let screenshot = image::open("screenshot.png").unwrap().to_luma8();
let matching_probability= imageproc::template_matching::match_template(&screenshot, &template, MatchTemplateMethod::CrossCorrelationNormalized);
let positions = find_extremes(&matching_probability);
println!("{:?}", positions);

极端值 { max_value： 0.9998113， min_value： 0.42247093， max_value_location： (843， 696)， min_value_location： (657， 832) }

我使用 ArrayFire 的示例

let template: Array<u8> = arrayfire::load_image(String::from("connect.png"), true);
let screenshot: Array<u8> = arrayfire::load_image(String::from("screenshot.png"), true);
let template_gray = rgb2gray(&template, 0.2126, 0.7152, 0.0722);
let screen_gray =  rgb2gray(&screenshot, 0.2126, 0.7152, 0.0722);
let matching_probability = arrayfire::match_template(&screen_gray, &template_gray, arrayfire::MatchType::LSAD);
af_print!("{:?}", matching_probability);

139569.0469 140099.2500 139869.8594 140015.7969 140680.9844

141952.5781 142602.7344 142870.7188...

从这里开始，我不知道如何获得最佳匹配的像素坐标。

Answer 1

Arrayfire不提供"极值"函数，而是将最小和最大函数族分开。

提供索引信息的那个以i为前缀。

imin_all和imax_all分别返回包装在 tupple 中的最小值和最大值索引。

您可以从值索引和数组维度中获取像素位置，因为知道 arrayfire 是列的主要。

let template: Array<u8> = arrayfire::load_image(String::from("connect.png"), true);
let screenshot: Array<u8> = arrayfire::load_image(String::from("screenshot.png"), true);
let template_gray = rgb2gray(&template, 0.2126, 0.7152, 0.0722);
let screen_gray = rgb2gray(&screenshot, 0.2126, 0.7152, 0.0722);
let matching_probability = arrayfire::match_template(&screen_gray, &template_gray, arrayfire::MatchType::LSAD);
let (min, _, min_idx) = imin_all(&matching_probability);
let (max, _, max_idx) = imax_all(&matching_probability);
let dims = matching_probability.dims();
let [_, height, _, _] = dims.get();
let px_x_min = min_idx as u64 / height;
let px_y_min = min_idx as u64 % height;
let px_x_max = max_idx as u64 / height;
let px_y_max = max_idx as u64 % height;
af_print!("{:?}", matching_probability);
println!("Minimum value: {} is at pixel ({},{}).",min, px_x_min, px_y_min);
println!("Maximum value: {} is at pixel ({},{}).", max, px_x_max, px_y_max);