如何找到图像中包含的图像？- AskGo

如何找到图像中包含的图像？

image-processing

opencv

python

我目前正在建立一个基本上相当于搜索引擎和网络漫画画廊之间的交叉点的网站，该画廊侧重于引用来源并给予作者以荣誉。

我正在尝试找出一种搜索图像以在其中查找字符的方法。

例如：

氰化物与幸福

假设我将红色字符和绿色字符另存为“ Red Man”和“ Green Man”，我如何确定图像是否包含一个或另一个。

不需要100％识别，或者什么是我想创建的更多功能，我不确定从哪里开始。我已经做了很多谷歌搜索的图像识别工作，但是并没有太大的帮助。

对于它的价值，我宁愿使用Python来完成。

参考资料：

Stack Overflow

2020-05-31 22:02:30 更新

花飞英• 612

共 2 个回答

高赞时间活跃

对于将来偶然发现此问题的任何人。

这可以通过模板匹配来完成。总而言之（我的理解），模板匹配在一个图像中寻找另一个图像中的精确匹配。

这是如何在Python中执行此操作的示例：

import cv2

method = cv2.TM_SQDIFF_NORMED

# Read the images from the file
small_image = cv2.imread('small_image.png')
large_image = cv2.imread('large_image.jpeg')

result = cv2.matchTemplate(small_image, large_image, method)

# We want the minimum squared difference
mn,_,mnLoc,_ = cv2.minMaxLoc(result)

# Draw the rectangle:
# Extract the coordinates of our best match
MPx,MPy = mnLoc

# Step 2: Get the size of the template. This is the same size as the match.
trows,tcols = small_image.shape[:2]

# Step 3: Draw the rectangle on large_image
cv2.rectangle(large_image, (MPx,MPy),(MPx+tcols,MPy+trows),(0,0,255),2)

# Display the original image with the rectangle around the match.
cv2.imshow('output',large_image)

# The image is only displayed if we call this
cv2.waitKey(0)

2020-05-31 22:04:08 更新

伯芦• 650

由于Moshe的答案仅涉及匹配模板，该模板在给定图片中仅包含一次。一次匹配多个的方法如下：

import cv2
import numpy as np

img_rgb = cv2.imread('mario.png')
template = cv2.imread('mario_coin.png')
w, h = template.shape[:-1]

res = cv2.matchTemplate(img_rgb, template, cv2.TM_CCOEFF_NORMED)
threshold = .8
loc = np.where(res >= threshold)
for pt in zip(*loc[::-1]):  # Switch collumns and rows
    cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0, 0, 255), 2)

cv2.imwrite('result.png', img_rgb)

（注意：我更改并修复了原始代码中的一些“错误” ）

结果：

检测马里奥硬币（之前/之后）