OpenCV 等效于 np.where()

Question

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

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

语法 loc[::-1] reverses数组。
zip(*loc..) 中的星号运算符正在用于解包给 zip 的 slice 。

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)

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

Answer 1

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

这在 golang 绑定(bind)中也可用。来自 gocv docs on FindNonZero :

func FindNonZero(src Mat, idx *Mat)

FindNonZero returns the list of locations of non-zero pixels.

For further details, please see: 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) 来表示点。