修改推理代码

69466de5 · liuhy · b44aeb9e · 69466de5
Commit 69466de5 authored Feb 27, 2023 by liuhy
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

LPRNet_ORT_infer.py LPRNet_ORT_infer.py +0 -1

No files found.
--- a/LPRNet_ORT_infer.py
+++ b/LPRNet_ORT_infer.py
@@ -44,7 +44,6 @@ def LPRNetInference(model, imgs):
    img = LPRNetPreprocess(imgs)
    
    if ort.get_device() == "GPU":
-        # sess = ort.InferenceSession(model, providers=['CUDAExecutionProvider'],) #GPU版本
        sess = ort.InferenceSession(model, providers=['ROCMExecutionProvider'],) #DCU版本
    else:
        sess = ort.InferenceSession(model, providers=['CPUExecutionProvider']) # CPU版本