Как распознать изображение независимо от его масштабаPython

Программы на Python
Anonymous
Как распознать изображение независимо от его масштаба

Сообщение Anonymous »

Изображения:
https://imgur.com/a/tzDGdyt
Крамка вокруг клубники
Означает, что клубника была обнаружена

Код: Выделить всё

import pyautogui
import cv2
import numpy
import time

time.sleep(2)

pyautogui.screenshot("screenshot.png")

Image2 = cv2.imread("screenshot.png")
Image1 = cv2.imread("berry.png")

Image2 = cv2.cvtColor(Image2, cv2.COLOR_BGR2GRAY)
Image1 = cv2.cvtColor(Image1, cv2.COLOR_BGR2GRAY)

result = cv2.matchTemplate(Image2, Image1, cv2.TM_CCOEFF_NORMED)

threshold = 0.7
locations = numpy.where(result >= threshold)

for point in zip(*locations[::-1]):
cv2.rectangle(Image2, point, (point[0] + Image1.shape[1], point[1] + Image1.shape[0]), (0, 255, 0), 2)

cv2.imshow('', Image2)
cv2.waitKey(0)
Цель: определить, находится ли изображение 1 внутри изображения 2.

Независимо от того, насколько масштабировано изображение 1.

Подробнее здесь: https://stackoverflow.com/questions/787 ... aled-it-is

Вернуться в «Python»