Как найти несколько одинаковых шаблонов OpenCV Python - PullRequest
Новая
       111

Как найти несколько одинаковых шаблонов OpenCV Python

0 голосов
/ 04 мая 2020

Извините за нубский вопрос. Моя проблема в том, что мультимасштаб находит только один из нескольких шаблонов в изображении, и мне нужны все из них. Как я могу это исправить? Я только новичок, буду благодарен за любую помощь или критику. 

for image in ims:

    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    found = None

    # loop over the scales of the image
    for scale in np.linspace(0.2, 1.0, 20)[::-1]:

        resized = imutils.resize(gray, width=int(gray.shape[1] * scale))
        r = gray.shape[1] / float(resized.shape[1])

        if resized.shape[0] < tH or resized.shape[1] < tW:
            break

        edged = cv2.Canny(resized, 150, 350)
        result = cv2.matchTemplate(edged, template, cv2.TM_CCOEFF)
        (_, maxVal, _, maxLoc) = cv2.minMaxLoc(result)


        clone = np.dstack([edged, edged, edged])
        cv2.rectangle(clone, (maxLoc[0], maxLoc[1]),
                      (maxLoc[0] + tW, maxLoc[1] + tH), (0, 0, 255), 2)

        if found is None or maxVal > found[0]:
            found = (maxVal, maxLoc, r)

    (_, maxLoc, r) = found
    (startX, startY) = (int(maxLoc[0] * r), int(maxLoc[1] * r))
    (endX, endY) = (int((maxLoc[0] + tW) * r), int((maxLoc[1] + tH) * r))
    if maxVal > 700000:
        cv2.rectangle(image, (startX, startY), (endX, endY), (0, 0, 255), 2)
        cv2.imshow('img',image)
        print('good')
    elif maxVal < 700000:
        print('bad')
print(maxVal)
cv2.waitKey(0)
cv2.destroyAllWindows()
...