From 724d5b21b75c5a5be09625a348c90c58186bb91b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 24 Jun 2022 14:18:58 +0200 Subject: [PATCH] `device_name="MYRIAD" for Intel NCS2` comment (#8327) Display device_name="MYRIAD" for Intel NCS2 option per user question in https://github.com/ultralytics/yolov5/issues/8154 --- models/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/common.py b/models/common.py index 66467a0ab1b7..7690f714def8 100644 --- a/models/common.py +++ b/models/common.py @@ -366,7 +366,7 @@ def __init__(self, weights='yolov5s.pt', device=torch.device('cpu'), dnn=False, if not Path(w).is_file(): # if not *.xml w = next(Path(w).glob('*.xml')) # get *.xml file from *_openvino_model dir network = ie.read_model(model=w, weights=Path(w).with_suffix('.bin')) - executable_network = ie.compile_model(model=network, device_name="CPU") + executable_network = ie.compile_model(network, device_name="CPU") # device_name="MYRIAD" for Intel NCS2 output_layer = next(iter(executable_network.outputs)) meta = Path(w).with_suffix('.yaml') if meta.exists():