基于pytorch框架和yolov5实现第一人称射击(FPS)游戏的辅助瞄准系统源码+项目说明.zip

import ttkbootstrap as ttk import tkinter as tk from ttkbootstrap.constants import * import json import os import time from ttkbootstrap.dialogs import Messagebox import keyboard winSizeX,winSizeY = 680,520 root = ttk.Window( title="FPS AI Aiming - Launcher", themename="superhero", size=(winSizeX,winSizeY), resizable=(False,False), ) root.place_window_center() id_game=ttk.IntVar() id_model=ttk.IntVar() id_runtype=ttk.IntVar() id_game.set(2) id_model.set(1) id_runtype.set(2) run_status=ttk.IntVar() run_status.set(0) num_mousesen=ttk.DoubleVar() num_mousesen.set(1.00) GAMES = [ ("CS 1.6", 1), ("CS : GO", 2), ("Apex Legends", 3)] MODELS = [ (" yolov5n ", 1), (" yolov5s ", 2), (" yolov5m ", 3)] RUNTYPE = [ (" Demo ", 1), (" Normal ", 2), (" Beta ", 3)] label_title = ttk.Label(root, text=" FPS AI AIMING", font=('Helvetica',30), bootstyle="light") label_title.pack(side=TOP, pady=(20,0)) label_version = ttk.Label(root,text=" v0.3.6\n20220715",bootstyle="secondary") label_version.place(x=winSizeX-100, y=winSizeY-60, anchor=NW) def show_running_warn(): print("show_info: ",Messagebox.show_info( message=" AI Aiming正在运行,改动将在下次启动生效", title="Warning",alert=False)) def Run(event=None): file_full_path = os.path.dirname(os.path.abspath(__file__)) if run_status.get()==0: params={'game':id_game.get(), 'model': id_model.get(),'runtype': id_runtype.get(), 'mouse_sen': num_mousesen.get()} filename=file_full_path+'\\bin\\launcher_params.json' with open(filename,'w') as file_obj: json.dump(params,file_obj) run_status.set(1) filename=file_full_path+'\\bin\\running_status.json' with open(filename,'w') as file_obj: json.dump(run_status.get(),file_obj) os.system('start pythonw '+file_full_path+'\\bin\\ai_0.3.2.py') os.system('start pythonw '+file_full_path+'\\bin\\asr_switch.py') if id_runtype.get() == 1: os.system('start pythonw '+file_full_path+'\\bin\\demo.py') btnRun.configure(text=" End the AI Aiming~ ", bootstyle="danger") root.title("FPS AI Aiming - Launcher [Running] ") if event is None: #按键启动 print("show_info: ",Messagebox.show_info( message=" AI-Aiming已启动!\n 首次使用或切换模型需下载模型文件,请稍等\n > 按【X】键自动瞄准并射击\n" +" > 点击启动器的End按钮或关闭启动器以结束AI-Aiming\n" +" > 如在CS:GO运行,请在游戏设置中关闭鼠标原始输入,\n打开鼠标加速并确保鼠标灵敏度与启动器保持一致", title="Tips",alert=False)) print(event) else: filename=file_full_path+'\\bin\\running_status.json' run_status.set(0) with open(filename,'w') as file_obj: json.dump(run_status.get(),file_obj) btnRun.configure(text=" Run the AI Aiming! ", bootstyle="secondary") root.title("FPS AI Aiming - Launcher") btnRun = ttk.Button(root, text=" Run the AI Aiming! ", bootstyle="secondary", command=Run) btnRun.pack(side=BOTTOM, pady=35) container4 = ttk.Frame(root) container4.place(x=30,y=350, anchor=NW) label_sign4 = ttk.Label(container4, text="Mouse sensitivity ", font=('Helvetica',12), bootstyle="secondary") label_sign4.pack(side=LEFT, pady=20, padx=20) def ms_onchange(input): # 滑动条移动时改变标签值 label_sign_ms_num.configure(text=str(num_mousesen.get())) if run_status.get()==1: show_running_warn() def ms_init(): # 滑动条初始化（窗口初始化时滑块移动到上次运行保存位置） try: file_full_path = os.path.dirname(os.path.abspath(__file__)) filename=file_full_path+'\\bin\\launcher_params.json' with open(filename, "r") as f: params = json.load(f) num_mousesen.set(params['mouse_sen']) ms_onchange(num_mousesen.get()) except: pass scale_ms=tk.Scale(container4, from_=0.05,to=8.00,orient="horizontal",length=262,resolution=0.01, command=ms_onchange,variable=num_mousesen ).pack(side=LEFT, padx=3) label_sign_ms_num = ttk.Label(container4, text=str(num_mousesen.get()), font=('Helvetica',12), bootstyle="secondary") label_sign_ms_num.pack(side=LEFT, pady=20, padx=3) container1 = ttk.Frame(root) container1.place(x=30,y=120, anchor=NW) def showScalems(): if id_game.get()==2: root.geometry(str(winSizeX)+"x"+str(winSizeY)) btnRun.pack(side=BOTTOM, pady=35) container4.place(x=30,y=350, anchor=NW) label_version.place(x=winSizeX-100, y=winSizeY-60, anchor=NW) else: root.geometry(str(winSizeX)+"x"+str(winSizeY-60)) btnRun.pack(side=BOTTOM, pady=35) container4.place_forget() label_version.place(x=winSizeX-100, y=winSizeY-120, anchor=NW) if run_status.get()==1: show_running_warn() label_sign1 = ttk.Label(container1, text="Choose the game ", font=('Helvetica',12), bootstyle="secondary") label_sign1.pack(side=LEFT, pady=20, padx=20) for games, num in GAMES: rbutton_game = ttk.Radiobutton(container1, text=games, variable=id_game, value=num, bootstyle="info-toolbutton", command=showScalems) rbutton_game.pack(side=LEFT, padx=3) container2 = ttk.Frame(root) container2.place(x=30,y=180, anchor=NW) def showModelWarn(): num = id_model.get() WARN = { 1 : " - Recommand at All Services", 2 : " - Recommand at RTX3050 and above", 3 : " - Recommand at RTX3080 and above" } label_warn.configure(text=WARN.get(num)) if run_status.get()==1: show_running_warn() label_sign2 = ttk.Label(container2, text="Choose the model ", font=('Helvetica',12), bootstyle="secondary") label_sign2.pack(side=LEFT, pady=20, padx=20) label_warn = ttk.Label(root, text=" - Recommand at All Services", font=('Helvetica',8), bootstyle="light") label_warn.place(x=290,y=245, anchor=NW) for models, num in MODELS: rbutton_model = ttk.Radiobutton(container2, text=models, variable=id_model, value=num, bootstyle="info-toolbutton", command=showModelWarn) rbutton_model.pack(side=LEFT, padx=3) container3 = ttk.Frame(root) container3.place(x=30,y=265, anchor=NW) def showModelWarn2(): num = id_runtype.get() WARN = { 1 : " - Show the AI detect process window", 2 : " - Run in normal mode", 3 : " - Try the beta function : dynamic prediction" } label_warn2.configure(text=WARN.get(num)) if run_status.get()==1: show_running_warn() label_sign3 = ttk.Label(container3, text="Choose the run-type", font=('Helvetica',12), bootstyle="secondary") label_sign3.pack(side=LEFT, pady=20, padx=20) for ltype, num in RUNTYPE: rbutton_runtype = ttk.Radiobutton(container3, text=ltype, variable=id_runtype, value=num, bootstyle="info-toolbutton", command=showModelWarn2) rbutton_runtype.pack(side=LEFT, padx=3) label_warn2 = ttk.Label(root, text=" - Run in normal mode", font=('Helvetica',8), bootstyle="light") label_warn2.place(x=290,y=330, anchor=NW) def close_window(): file_full_path = os.path.dirname(os.path.abspath(__file__)) filename=file_full_path+'\\bin\\running_status.json' run_status.set(0) with open(filename,'w') as file_obj: json.dump(run_status.get(),file_obj) os._exit(0) ms_init() # root.bind('<Shift-Control-KeyPress-P>',param_panel) root.bind("<Shift-Control-KeyPress-R>",Run) #使用快捷键运行将不显示弹窗