pull/17463/head^2
UltralyticsAssistant 1 month ago
parent 586cc53e39
commit 445c7e6938
  1. 2
      ultralytics/nn/autobackend.py

@ -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

Loading…
Cancel
Save