选择图像内部的图像

Question

我有这个图像，我想选择其中的所有地图，包括使用python的名称。

我用OpenCv试过了

from cv2 import cv2
import numpy as np
import os

for image in os.listdir('assets'):
    print(image)
    img_rgb = cv2.imread(image)
    template = cv2.imread('assets/template/template.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 columns and rows
        cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0, 0, 255), 2)

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

但是我总是得到这个错误：

OpenCV（4.2.0）C:\projects\OpenCV python\OpenCV\modules\imgproc\src\templammatch。cpp:1104：错误：（-215：断言失败）（深度==CV\U 8U | |深度==CV\U 32F）

我该怎么做？请帮帮忙

Answer 1

<代码>cv2。当matchTemplate的一个参数为NoneType时，matchTemplate将返回此错误。换句话说，img\U rgb什么都不是，很可能是因为在该位置没有找到图像。

以下是我如何再现您的错误：

from cv2 import cv2
import numpy as np

img_rgb = cv2.imread('fjdlfkjdlkfjdlfksdad')  # or None
template = np.random.randint(0, 256, (100, 100, 3))
res = cv2.matchTemplate(img_rgb, template, cv2.TM_CCOEFF_NORMED)

cv2.error：（-215：断言失败）（深度==CV_8U||深度==CV_32F）