|
|
@ -179,7 +179,6 @@ class AutoBackend(nn.Module): |
|
|
|
# providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] if cuda else ["CPUExecutionProvider"] |
|
|
|
# providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] if cuda else ["CPUExecutionProvider"] |
|
|
|
providers = ["CPUExecutionProvider"] |
|
|
|
providers = ["CPUExecutionProvider"] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if not cuda and "CUDAExecutionProvider" in providers: |
|
|
|
if not cuda and "CUDAExecutionProvider" in providers: |
|
|
|
providers.remove("CUDAExecutionProvider") |
|
|
|
providers.remove("CUDAExecutionProvider") |
|
|
|
elif cuda and "CUDAExecutionProvider" not in providers: |
|
|
|
elif cuda and "CUDAExecutionProvider" not in providers: |
|
|
@ -521,6 +520,7 @@ class AutoBackend(nn.Module): |
|
|
|
y = self.bindings |
|
|
|
y = self.bindings |
|
|
|
if self.mct: |
|
|
|
if self.mct: |
|
|
|
from ultralytics.utils.ops import xyxy2xywh |
|
|
|
from ultralytics.utils.ops import xyxy2xywh |
|
|
|
|
|
|
|
|
|
|
|
y = np.concatenate([xyxy2xywh(y[0]), y[1]], axis=-1).transpose(0, 2, 1) |
|
|
|
y = np.concatenate([xyxy2xywh(y[0]), y[1]], axis=-1).transpose(0, 2, 1) |
|
|
|
|
|
|
|
|
|
|
|
# OpenVINO |
|
|
|
# OpenVINO |
|
|
|