自瞄

自瞄框框可新开个线程，然后启动这个即可，有点耗内存，录像不知为啥显示不出来。

def draw():
    global data
    while True:
        for i in range(xlen):
                win32gui.SetPixel(dc, x+i, y, 255)
                win32gui.SetPixel(dc, x+i, y+ylen, 255)
        for i in range(ylen):
            win32gui.SetPixel(dc, x, y+i, 255)
            win32gui.SetPixel(dc, x+xlen, y+i, 255)
        for man in data:
            _x = int(x+man[0].item())
            _y = int(y+man[1].item())
            _xlen = int((man[2].item()-man[0].item())/2)
            _ylen = int((man[3].item()-man[1].item())/2)
            _x = int(_x+_xlen/2)
            _y = int(_y+_ylen/2)
            for i in range(_xlen):
                win32gui.SetPixel(dc, _x+i, _y, 255)
                win32gui.SetPixel(dc, _x+i, _y+_ylen, 255)
            for i in range(_ylen):
                win32gui.SetPixel(dc, _x, _y+i, 255)
                win32gui.SetPixel(dc, _x+_xlen, _y+i, 255)

自瞄代码

import torch
import pyautogui
import win32gui
import sys
import threading
import time
from pynput import keyboard
dc = win32gui.GetDC(0)
x=300 #起点x位置
y=274 #起点y位置
xlen = 424#边框长度
ylen = 268
star_x =x + int(xlen/2)#准星位置
star_y= y + int(ylen/2)
data = torch.Tensor([])
device = torch.device('cuda')
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
model.classes=[0]
def shot():
    a = time.time()
    img = pyautogui.screenshot(region=[x,y,xlen,ylen])
    results = model(img)
    data = (results.xyxy[0]).clone().type(torch.int)
    if len(data) !=0:
        item_x = 0
        item_y = 0
        min = sys.maxsize 
        for man in data:
            _x = x+man[0].item()
            _y = y+man[1].item()
            _xlen = man[2].item()-man[0].item()
            _ylen = man[3].item()-man[1].item()
            _x = _x+_xlen/2
            _y = _y+_ylen/4
            if ((_x-star_x)*(_x-star_x) + (_y-star_y)*(_y-star_y))