放弃机器学习，Python 如何进行物体检测？

Posted 2021-04-06 智能卫星研究院

每当听到“物体检测”时，就会想到机器学习和各种不同的框架。但实际上，我们可以在不使用机器学习或任何其他框架的情况下进行物体检测。本文将会展示如何仅使用Python进行操作。

首先，我们定义一个模板图像（或者叫模板物体），然后程序将在源图像中查找与我们选择的模板匹配的所有其他物体。举例来说明一下。下面有两张图片，上面是飞机的源图像，下面是模板照片，其中的物体为飞机。

下面我们来编写python代码，圈出源图像中所有匹配模板图像的区域。

首先，我们来检测一个物体。然后再调整代码实现多个物体的检测。

 

检测一个物体：最准确的那个物体

 

我们需要一个源图像和一个模板图像。模板图像在源图像上滑动（像2D卷积有一样），然后程序将尝试找到最准确的匹配项。

下面我们开始写代码。

import cv2import numpy as npfrom matplotlib import pyplot as pltimg_rgb = cv2.imread('SourceIMG.jpeg')img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)template = cv2.imread('TemplateIMG.jpeg', 0)

在上述代码中，我们使用OpenCV读取SourceIMG.jpeg和TemplateIMG.jpeg。

height, width = template.shape[::]

模板图像会在整个源图像上滑动，对整个区域进行搜索（将左上角作为参考框）。模板图像与源图像匹配后，记下左上角的位置，然后在实际匹配的区域周围绘制一个框。为此，我们需要知道此模板图像的高度和宽度。下面来绘制矩形。

res = cv2.matchTemplate(img_gray, template, cv2.TM_SQDIFF)

模板匹配是OpenCV提供的功能，它利用源图像和模板图像的灰度图像，计算我们需要的统计指标。这里使用的是最小平方差（TM_SQDIFF），因为寻找的是模板图像和源图像之间的最小差。

plt.imshow(res, cmap='gray')

如果将到目前为止的结果绘制成图，就会得到一个概率图。从下图可以看到，这些小点是模板实际匹配的位置。

min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res)

我们可以使用上面的代码从概率图中找出小点的位置。然后使用minMaxLoc(res)提取最小值、最大值、最小值的位置和最大值的位置。

top_left = min_loc bottom_right = (top_left[0] + width, top_left[1] + height)cv2.rectangle(img_rgb, top_left, bottom_right, (255, 0, 0), 2)

为了在模板图像匹配的源图像上绘制一个蓝色矩形，我们需要获得最小值的位置min_loc（该位置为匹配开始的位置）作为左上角。同样，可以通过top_left[0]+width和top_left[1]+height获得右下角。通过这些尺寸，使用cv2.rectangle绘制蓝色矩形。

一切准备就绪，进行可视化。

cv2.imshow("Matched image", img_rgb)cv2.waitKey()cv2.destroyAllWindows()

完整的代码：

import cv2import numpy as npfrom matplotlib import pyplot as plt
img_rgb = cv2.imread('SourceIMG.jpeg')img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)template = cv2.imread('TemplateIMG.jpeg', 0)
height, width = template.shape[::]
res = cv2.matchTemplate(img_gray, template, cv2.TM_SQDIFF)plt.imshow(res, cmap='gray')
min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res)
top_left = min_loc #Change to max_loc for all except for TM_SQDIFFbottom_right = (top_left[0] + width, top_left[1] + height)cv2.rectangle(img_rgb, top_left, bottom_right, (255, 0, 0), 2)
cv2.imshow("Matched image", img_rgb)cv2.waitKey()cv2.destroyAllWindows()

检测多个物体：在给定阈值下进行检测

 

以上已经完成了单个物体的检测，即选择源图像和模板图像之差的最小值。通过定义阈值的方法，我们可以检测所有与模板图像相似的物体。

为此，将使用与上例相同的源图像和模板图像，并设置阈值为概率大于0.5（你可以查看res数组来确定阈值）。只需要更改几行代码即可检测多个物体。

res = cv2.matchTemplate(img_gray, template, cv2.TM_CCOEFF_NORMED)

在这里，使用TM_CCOEFF_NORMED，因为需要获取最大值，而不是最小值。这意味着需要寻找多个物体而不是一个。

threshold = 0.5 #For TM_CCOEFF_NORMED, larger values means good fitloc = np.where( res >= threshold)

我们要查找所有大于阈值的位置值。loc接收2个输出数组，并将这些数组组合在一起，这样就可以获得x，y坐标。

for pt in zip(*loc[::-1]): cv2.rectangle(img_rgb, pt, (pt[0]+width, pt[1]+height), (255, 0, 0), 1)

这里有多个位置。因此，需要针对所有位置绘制蓝色矩形。下面来进行可视化。

完整的代码：

import cv2import numpy as npfrom matplotlib import pyplot as plt
img_rgb=cv2.imread('SourceIMG.jpeg')img_gray=cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)template=cv2.imread('TemplateIMG.jpeg', 0)
height, width=template.shape[::]
res=cv2.matchTemplate(img_gray, template, cv2.TM_CCOEFF_NORMED)plt.imshow(res, cmap='gray')
threshold=0.5 #For TM_CCOEFF_NORMED, larger values = good fit.
loc=np.where(res>=threshold)  
for pt in zip(loc[::-1]):  cv2.rectangle(img_rgb, pt, (pt[0]+width, pt[1]+height), (255, 0, 0), 1) 
cv2.imshow("Matched image", img_rgb)cv2.waitKey()cv2.destroyAllWindows()

看起来很简单吧？但是如果使用机器学习或框架，则可以达到更高的准确性。

感谢您的阅读，希望本文对您有所帮助。

原文：https://towardsdatascience.com/object-detection-on-python-using-template-matching-ab4243a0ca62

文章来源

作者 | Ravindu Senaratne

译者 | 弯月，责编 | 屠敏

头图 | CSDN 下载自视觉中国
出品 | CSDN（ID：CSDNnews）

智能卫星研究院