mirror of
https://github.com/RootKit-Org/AI-Aimbot.git
synced 2025-06-21 02:41:01 +08:00
Update main_onnx_cpu.py
This commit is contained in:
parent
31246d927c
commit
02a9aee048
@ -117,7 +117,7 @@ def main():
|
||||
if len(im.shape) == 3:
|
||||
im = im[None]
|
||||
|
||||
outputs = ort_sess.run(None, {'images': npImg})
|
||||
outputs = ort_sess.run(None, {'images': np.array(im)})
|
||||
|
||||
im = torch.from_numpy(outputs[0]).to('cpu')
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user