《python + opencv实现目标检测》

pip install numpy
pip install opencv-python

python
>>> import cv2

pip install datetime
pip install pygame
pip install keyboard
pip install pywin32
pip install pyautogui
pip install Pillow

# -*- coding: utf-8 -*-
import cv2
import numpy as np
import time
import pyautogui
import keyboard
import sys
import pygame
import datetime
import os

#窗口位置
game_rect = (0, 0, 800, 600)  

#需要检测的目标
myself_template = cv2.imread('myself.png', cv2.IMREAD_GRAYSCALE)
wheel_template = cv2.imread('wheel.png', cv2.IMREAD_GRAYSCALE)
other_template = cv2.imread('other.png', cv2.IMREAD_GRAYSCALE)

#check_green_template = cv2.imread('check_green.png', cv2.IMREAD_GRAYSCALE)
player_time = 0

#音乐相关初始化
pygame.init()
pygame.mixer.music.load('childhood.mp3')


#切换窗口
def exit_game():
    keyboard.press('alt')
    time.sleep(0.1)
    keyboard.press('tab')
    time.sleep(0.1)
    keyboard.release('alt')
    time.sleep(0.1)
    keyboard.release('tab')
    #退出整个程序
    sys.exit()

def play_audio():
    if not pygame.mixer.music.get_busy():
        pygame.mixer.music.play()

while True:
    #屏幕截图加载图片
    screenshot = pyautogui.screenshot(region=game_rect)
    screenshot = np.array(screenshot)
    screenshot = screenshot[:, :, ::-1].copy()
    screenshot_gray = cv2.cvtColor(screenshot, cv2.COLOR_BGR2GRAY)

    #获取当前时间
    current_time = datetime.datetime.now()
    print(f"[{current_time}] ")


    #轮检测
    wheel_result = cv2.matchTemplate(screenshot_gray, wheel_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.99
    hole_loc = np.where(wheel_result >= threshold)
    num_matches = len(hole_loc[0])
    if num_matches > 0:
        print(u"检测到地图轮个数为:",f"{num_matches}")
    '''if num_matches == 1:
        exit_game() '''          

    #测谎检测
    '''check_green_result = cv2.matchTemplate(screenshot_gray, check_green_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.99
    other_loc = np.where(check_green_result >= threshold)
    num_matches = len(other_loc[0])
    if num_matches > 0:
        print(u"检测到测谎仪:",f"{num_matches}")'''
    for filename in os.listdir('check_folder'):
        if filename.endswith('.jpg') or filename.endswith('.png'):
            check_img = cv2.imread(os.path.join('check_folder', filename), cv2.IMREAD_GRAYSCALE)
            check_result = cv2.matchTemplate(screenshot_gray, check_img, cv2.TM_CCOEFF_NORMED)
            threshold = 0.97
            other_loc = np.where(check_result >= threshold)
            num_matches = len(other_loc[0])
            if num_matches > 0:
                print(u"检测到测谎仪:",f"{num_matches}")
                play_audio()
                

    #其他玩家检测
    other_result = cv2.matchTemplate(screenshot_gray, other_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.99
    other_loc = np.where(other_result >= threshold)
    num_matches = len(other_loc[0])
    if num_matches > 0:
        player_time += 1;
        if player_time > 5:
            player_time = 0
            print(u"检测到其他人进入地图人数为:",f"{num_matches}")
            play_audio()
    else:
        player_time = 0


    #自己检测
    myself_result = cv2.matchTemplate(screenshot_gray, myself_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.99
    myself_loc = np.where(myself_result >= threshold)
    num_matches = len(myself_loc[0])
    if num_matches > 0:
        print(u"检测到我自己人数:",f"{num_matches}") 

    time.sleep(1)