IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix #38545 remove CPU network layer check for py samples (#2145)

Browse Source
This commit is contained in:
Kate Generalova 2020-09-10 12:44:13 +03:00 committed by GitHub
parent 926be83568
commit d4d460101d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 0 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
inference-engine/ie_bridges/python/sample/classification_sample/classification_sample.py
View File

@ -63,18 +63,6 @@ def main():
# Read IR # Read IR
log.info("Loading network files:\n\t{}\n\t{}".format(model_xml, model_bin)) log.info("Loading network files:\n\t{}\n\t{}".format(model_xml, model_bin))
net = ie.read_network(model=model_xml, weights=model_bin) net = ie.read_network(model=model_xml, weights=model_bin)
func = ng.function_from_cnn(net)
if "CPU" in args.device:
supported_layers = ie.query_network(net, "CPU")
ops = func.get_ordered_ops()
not_supported_layers = [op.friendly_name for op in ops if op.friendly_name not in supported_layers]
if len(not_supported_layers) != 0:
log.error("Following layers are not supported by the plugin for specified device {}:\n {}".
format(args.device, ', '.join(not_supported_layers)))
log.error("Please try to specify cpu extensions library path in sample's command line parameters using -l "
"or --cpu_extension command line argument")
sys.exit(1)
assert len(net.input_info.keys()) == 1, "Sample supports only single input topologies" assert len(net.input_info.keys()) == 1, "Sample supports only single input topologies"
assert len(net.outputs) == 1, "Sample supports only single output topologies" assert len(net.outputs) == 1, "Sample supports only single output topologies"

11
inference-engine/ie_bridges/python/sample/classification_sample_async/classification_sample_async.py
View File

@ -108,18 +108,7 @@ def main():
# Read IR # Read IR
log.info("Loading network files:\n\t{}\n\t{}".format(model_xml, model_bin)) log.info("Loading network files:\n\t{}\n\t{}".format(model_xml, model_bin))
net = ie.read_network(model=model_xml, weights=model_bin) net = ie.read_network(model=model_xml, weights=model_bin)
func = ng.function_from_cnn(net)
if "CPU" in args.device:
supported_layers = ie.query_network(net, "CPU")
ops = func.get_ordered_ops()
not_supported_layers = [op.friendly_name for op in ops if op.friendly_name not in supported_layers]
if len(not_supported_layers) != 0:
log.error("Following layers are not supported by the plugin for specified device {}:\n {}".
format(args.device, ', '.join(not_supported_layers)))
log.error("Please try to specify cpu extensions library path in sample's command line parameters using -l "
"or --cpu_extension command line argument")
sys.exit(1)
assert len(net.input_info.keys()) == 1, "Sample supports only single input topologies" assert len(net.input_info.keys()) == 1, "Sample supports only single input topologies"
assert len(net.outputs) == 1, "Sample supports only single output topologies" assert len(net.outputs) == 1, "Sample supports only single output topologies"

10
inference-engine/ie_bridges/python/sample/object_detection_sample_ssd/object_detection_sample_ssd.py
View File

@ -73,16 +73,6 @@ def main():
if args.cpu_extension and "CPU" in args.device: if args.cpu_extension and "CPU" in args.device:
ie.add_extension(args.cpu_extension, "CPU") ie.add_extension(args.cpu_extension, "CPU")
log.info("CPU extension loaded: {}".format(args.cpu_extension)) log.info("CPU extension loaded: {}".format(args.cpu_extension))
if "CPU" in args.device:
supported_layers = ie.query_network(net, "CPU")
not_supported_layers = [op.friendly_name for op in ops if op.friendly_name not in supported_layers]
if len(not_supported_layers) != 0:
log.error("Following layers are not supported by the plugin for specified device {}:\n {}".
format(args.device, ', '.join(not_supported_layers)))
log.error("Please try to specify cpu extensions library path in sample's command line parameters using -l "
"or --cpu_extension command line argument")
sys.exit(1)
# ----------------------------------------------------------------------------------------------------- # -----------------------------------------------------------------------------------------------------
# --------------------------- 3. Read and preprocess input -------------------------------------------- # --------------------------- 3. Read and preprocess input --------------------------------------------

12
inference-engine/ie_bridges/python/sample/style_transfer_sample/style_transfer_sample.py
View File

@ -67,18 +67,6 @@ def main():
# Read IR # Read IR
log.info("Loading network files:\n\t{}\n\t{}".format(model_xml, model_bin)) log.info("Loading network files:\n\t{}\n\t{}".format(model_xml, model_bin))
net = ie.read_network(model=model_xml, weights=model_bin) net = ie.read_network(model=model_xml, weights=model_bin)
func = ng.function_from_cnn(net)
if "CPU" in args.device:
supported_layers = ie.query_network(net, "CPU")
ops = func.get_ordered_ops()
not_supported_layers = [op.friendly_name for op in ops if op.friendly_name not in supported_layers]
if len(not_supported_layers) != 0:
log.error("Following layers are not supported by the plugin for specified device {}:\n {}".
format(args.device, ', '.join(not_supported_layers)))
log.error("Please try to specify cpu extensions library path in sample's command line parameters using -l "
"or --cpu_extension command line argument")
sys.exit(1)
assert len(net.input_info.keys()) == 1, "Sample supports only single input topologies" assert len(net.input_info.keys()) == 1, "Sample supports only single input topologies"
assert len(net.outputs) == 1, "Sample supports only single output topologies" assert len(net.outputs) == 1, "Sample supports only single output topologies"