时间戳

首页 > 解决方案 > OpenCV 等效于 np.where()

opencv - OpenCV 等效于 np.where()

问题描述

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

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

语法loc[::-1] 反转数组。

星号运算符 inzip(*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)

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

标签： opencvgomatrixcomputer-visiongocv

解决方案

OpenCV 有一个内置的（半）等价函数np.where()，即findNonZero(). 正如名称所暗示的那样，它会在图像中找到非零元素，这就是np.where()使用单个参数调用时所做的事情，如numpy 文档所述。

这在 golang 绑定中也可用。从gocvFindNonZero 上的文档：

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) 作为点。

推荐阅读