OpenVINO metadata fix2 (#7954)

* Bug Fixed: OpenVINO metadata

* Bug Fixed: OpenVINO metadata

* Update export.py

* Update export.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
xylieongandGlenn Jocher authored and GitHub committed 2022-05-24 13:19:25 +02:00
1 parent 0dd66e2dc7
commit 27911dc824
2 files changed
+2 -2

No files matched your search

+1 -1
View File
@@ -368,7 +368,7 @@ class DetectMultiBackend(nn.Module):
network = ie.read_model(model=w, weights=Path(w).with_suffix('.bin'))
executable_network = ie.compile_model(model=network, device_name="CPU")
output_layer = next(iter(executable_network.outputs))
meta = w.with_suffix('.yaml')
meta = Path(w).with_suffix('.yaml')
if meta.exists():
stride, names = self._load_metadata(meta) # load metadata
elif engine: # TensorRT