OpenCV 等效于 np.where()

Question

当使用gocv 包时，例如，可以对图像中的图案执行template matching。该包还提供了MinMaxLoc 函数来检索矩阵中的最小值和最大值的位置。

但是，在下面的 python 示例中，作者使用numpy.Where 对矩阵进行阈值处理并获取多个最大值的位置。 python zip 函数用于将值粘合在一起，因此它们就像一个切片 [][2]int，内部切片是找到的匹配项的 xs 和 ys。

语法loc[::-1]reverses数组。

zip(*loc..) 中的星号运算符用于解包给 zip 的切片。

https://docs.opencv.org/master/d4/dc6/tutorial_py_template_matching.html

import cv2 as cv
import numpy as np
from matplotlib import pyplot as plt

img_rgb = cv.imread('mario.png')
img_gray = cv.cvtColor(img_rgb, cv.COLOR_BGR2GRAY)
template = cv.imread('mario_coin.png',0)
w, h = template.shape[::-1]
res = cv.matchTemplate(img_gray,template,cv.TM_CCOEFF_NORMED)
threshold = 0.8
loc = np.where( res >= threshold)

for pt in zip(*loc[::-1]):
    cv.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0,0,255), 2)
cv.imwrite('res.png',img_rgb)

如何在 Go 中实现相同的 np.where 算法以在应用阈值后获取多个位置？

Answer 1

OpenCV 有一个内置的（半）等价于np.where() 的函数，即findNonZero()。顾名思义，它会在图像中找到非零元素，这就是 np.where() 在使用单个参数调用时所做的事情，如 the numpy docs 状态。

这在 golang 绑定中也可用。来自gocv docs on FindNonZero：

func FindNonZero(src Mat, idx *Mat)

FindNonZero 返回非零像素的位置列表。

更多详情请见：https://docs.opencv.org/master/d2/de8/group__core__array.html#gaed7df59a3539b4cc0fe5c9c8d7586190

注意：np.where() 按数组顺序返回索引，即 (row, col) 或 (i, j)，这与典型的图像索引 (x, y) 相反。这就是为什么 loc 在 Python 中被反转的原因。当使用findNonZero() 时，您不需要这样做，因为 OpenCV 总是使用 (x, y) 作为点。