主页 / 问题 / 1003359
Accepted
rybic
Asked:2020-07-16 18:38:03 +0000 UTC2020-07-16 18:38:03 +0000 UTC

如何处理实时屏幕图像，本质上是做视频截屏

  • 772

我需要从监视器屏幕上获取图像，以便实时更新图像（然后我将搜索指定的图像（对象），我已经实现了搜索）。如何在 python 3.x 中使用哪些库来做到这一点？

搜索代码：

import cv2 
import numpy as np

img = cv2.imread("screen0.png")  #работает только с 1 картинкой
imgg = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
template = cv2.imread("stone.png", cv2.IMREAD_GRAYSCALE)
w, h = template.shape[::-1]

result = cv2.matchTemplate(imgg, template, cv2.TM_CCOEFF_NORMED)
loc = np.where(result >= 0.5)

for pt in zip(*loc[::-1]):
    cv2.rectangle(img, pt, (pt[0] + w, pt[1] + h), (0, 255, 0), 3)

cv2.imshow("img", img)
python-3.x

1 个回答

  1. Best Answer
    from time import sleep
import numpy
from PIL import ImageGrab
import cv2


class Main:
    def __init__(self):

        self.template = cv2.imread("template.png", cv2.IMREAD_GRAYSCALE)

        self.width, self.height = self.template.shape[::-1]

    def run(self):
        while True:

            sleep(1)

            array = cv2.cvtColor(numpy.array(ImageGrab.grab()), cv2.COLOR_RGB2BGR)

            screen_shot_gray = cv2.cvtColor(array, cv2.COLOR_BGR2GRAY)

            result = cv2.matchTemplate(screen_shot_gray, self.template, cv2.TM_CCOEFF_NORMED)

            loc = numpy.where(result >= 0.5)

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

            print(result)
            # Здесь добавляете код pyautogui

if __name__ == '__main__':
    main = Main()
    main.run()
    • 0

相关问题