Fix OpenVINO Usage example (#9874)
* Fix OpenVINO Usage example * Fix OpenVINO Usage example
This commit is contained in:
1 parent
6371de8879
commit
3b1a9d22a4
8 files changed
+10
-10
No files matched your search
+2
-2
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user