Доброго времени суток, мудрые пользователи Хабра.
Недавно я наткнулся на видео об автоматизации рыбалки в ММО Albion Online /
https://www.youtube.com/watch?v=92XnJ3s0vX8 / так же, есть статья на Хабре об этом. Так как я играю в эту игру я решил написать свой скрипт, определяющий красные ники(врагов) на краях экрана и нажимающий кнопку R, с помощью библиотек OpenCV и PyAutoGUI, это такая сфера игры, как ганг, но не будем вдаваться в подробности.
Исходный код -
spoilerimport time
import cv2
import mss
import numpy as np
import pyautogui
template = cv2.imread("red_nick.png", cv2.IMREAD_GRAYSCALE)
w, h = template.shape[::-1]
with mss.mss() as sct:
monitor = {"top": 0, "left": 0, "width": 400, "height": 250}
while "Screen capturing":
last_time = time.time()
img = np.array(sct.grab(monitor))
empty_img = np.array(sct.grab(monitor))
print("fps: {}".format(1 / (time.time() - last_time)))
gray_frame = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
res = cv2.matchTemplate(gray_frame, template, cv2.TM_CCOEFF_NORMED)
loc = np.where(res >= 0.999)
for i in zip(*loc[::-1]):
cv2.rectangle(img, i, (i[0] + w, i[1] + h), (0, 255, 255), 10)
#cv2.imshow("Frame", img)
#cv2.imshow("Frame2", empty_img)
if cv2.waitKey(1) == 27:
cv2.destroyAllWindows()
break
if (img != empty_img).any():
pyautogui.press(keys='r')
cv2.imshow("Frame", img)
cv2.waitKey()
break
Проблема сразу в нескольких вещах, начнем по порядку:
1) Я взял за основу код из статьи на Хабре, т.к. сам сейчас не сильно разбираюсь, поэтому код может быть в некоторых местах кривоват.
2) Вторая проблема в этих строчках кода -
if (img != empty_img).any():
pyautogui.press(keys='r')
break
Во первых, сделано на коленке, чтобы хотя бы не выдавало ошибку при запуске.
Во вторых, это задумывалось так, чтобы в переменную img записывался скриншот экрана(многоразово, чтобы было как видео), а в empty_img - тот же самый скриншот, но без изменений. То есть, находя с помощью OpenCV красный ник на img, мы выделяем его и сравниваем img с empty_img, чтобы узнать, если ли разница между ними(есть ли на скриншоте красный ник). И затем, если мы находили это - нажимали кнопку R, если нет - продолжали цикл. Но это не работает так, как задумывалось, потому что код иногда может просто не успеть записать в эти две переменные два одинаковых скриншота, что мешает его правильному выполненению. Я попробовал решить эту проблему, заменив
empty_img = np.array(sct.grab(monitor))
на
empty_img = img,
что, логически, должно было помочь, но возникла другая проблема - после выполнения строк
for i in zip(*loc[::-1]):
cv2.rectangle(img, i, (i[0] + w, i[1] + h), (0, 255, 255), 10)
эти две переменные становятся одинаковыми. Почему, я так и не понял.
Исходный код -
spoilerimport time
import cv2
import mss
import numpy as np
import pyautogui
template = cv2.imread("red_nick.png", cv2.IMREAD_GRAYSCALE)
w, h = template.shape[::-1]
with mss.mss() as sct:
monitor = {"top": 0, "left": 0, "width": 400, "height": 250}
while "Screen capturing":
last_time = time.time()
img = np.array(sct.grab(monitor))
empty_img = np.array(sct.grab(monitor))
print("fps: {}".format(1 / (time.time() - last_time)))
gray_frame = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
res = cv2.matchTemplate(gray_frame, template, cv2.TM_CCOEFF_NORMED)
loc = np.where(res >= 0.999)
for i in zip(*loc[::-1]):
cv2.rectangle(img, i, (i[0] + w, i[1] + h), (0, 255, 255), 10)
#cv2.imshow("Frame", img)
#cv2.imshow("Frame2", empty_img)
if cv2.waitKey(1) == 27:
cv2.destroyAllWindows()
break
if (img != empty_img).any():
pyautogui.press(keys='r')
cv2.imshow("Frame", img)
cv2.waitKey()
break
Прошу вас, друзья, помочь мне либо исправить данную версию решения этой задачи, либо придумать решение лучше этого, работающее без заморочек(или с ними :D). Заранее спасибо всем, кто откликнется!
Простой
