barian0517/APEX_AIMBOT
1
0
Fork 0
mirror of https://github.com/RootKit-Org/AI-Aimbot.git synced 2025-06-21 02:41:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed The AMD_GPU file to not import cuda and added the forgotton torch DML import (#63)

Browse Source 
* Update main_onnx_amd.py

* Update main_onnx_amd.py
This commit is contained in:
Villageslayer 2023-03-27 06:07:28 +02:00 committed by GitHub
parent 8e6d0d3ac7
commit bc9af95607
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main_onnx_amd.py
View File

@ -1,6 +1,5 @@
import onnxruntime as ort
import numpy as np
import cupy as cp
import pyautogui
import gc
import numpy as np
@ -12,6 +11,7 @@ import pandas as pd
from utils.general import (cv2, non_max_suppression, xyxy2xywh)
import dxcam
import torch
import torch_directml
def main():
@ -109,7 +109,7 @@ def main():
npImg = npImg.astype(cp.half)
npImg = cp.moveaxis(npImg, 3, 1)
outputs = ort_sess.run(None, {'images': cp.asnumpy(npImg)})
outputs = ort_sess.run(None, {'images': npImg})
im = torch.from_numpy(outputs[0]).to('cpu')