Fix OpenVINO Usage example (#9874)

* Fix OpenVINO Usage example

* Fix OpenVINO Usage example
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-10-20 19:54:07 +02:00
1 parent 6371de8879
commit 3b1a9d22a4
8 files changed
+10 -10

No files matched your search

+2 -2
View File
@@ -318,7 +318,7 @@ class DetectMultiBackend(nn.Module):
# TorchScript: *.torchscript
# ONNX Runtime: *.onnx
# ONNX OpenCV DNN: *.onnx --dnn
# OpenVINO: *.xml
# OpenVINO: *_openvino_model
# CoreML: *.mlmodel
# TensorRT: *.engine
# TensorFlow SavedModel: *_saved_model
@@ -469,7 +469,7 @@ class DetectMultiBackend(nn.Module):
check_requirements('paddlepaddle-gpu' if cuda else 'paddlepaddle')
import paddle.inference as pdi
if not Path(w).is_file(): # if not *.pdmodel
w = next(Path(w).rglob('*.pdmodel')) # get *.xml file from *_openvino_model dir
w = next(Path(w).rglob('*.pdmodel')) # get *.pdmodel file from *_paddle_model dir
weights = Path(w).with_suffix('.pdiparams')
config = pdi.Config(str(w), str(weights))
if cuda: