def test_read_net_from_buffer(): ie = IECore() with open(test_net_bin, 'rb') as f: bin = f.read() with open(model_path()[0], 'rb') as f: xml = f.read() net = ie.read_network(model=xml, weights=bin, init_from_buffer=True) assert isinstance(net, IENetwork)
def test_net_from_buffer_valid(): ie = IECore() with open(test_net_bin, 'rb') as f: bin = f.read() with open(model_path()[0], 'rb') as f: xml = f.read() net = ie.read_network(model=xml, weights=bin, init_from_buffer=True) net2 = ie.read_network(model=test_net_xml, weights=test_net_bin) for name, layer in net.layers.items(): for blob, data in layer.blobs.items(): assert np.allclose(data, net2.layers[name].blobs[blob]), \ "Incorrect weights for layer {} and blob {}".format(name, blob)
def test_net_from_buffer_valid_deprecated(): ie = IECore() with open(test_net_bin, 'rb') as f: bin = f.read() with open(model_path()[0], 'rb') as f: xml = f.read() net = ie.read_network(model=xml, weights=bin, init_from_buffer=True) ref_net = ie.read_network(model=test_net_xml, weights=test_net_bin) assert net.name == ref_net.name assert net.batch_size == ref_net.batch_size ii_net = net.input_info ii_net2 = ref_net.input_info o_net = net.outputs o_net2 = ref_net.outputs assert ii_net.keys() == ii_net2.keys() assert o_net.keys() == o_net2.keys()
import warnings import numpy from openvino.inference_engine import DataPtr, IECore from conftest import model_path test_net_xml, test_net_bin = model_path() def test_name(recwarn): warnings.simplefilter("always") ie = IECore() net = ie.read_network(model=test_net_xml, weights=test_net_bin) assert net.layers['27'].name == "27" assert len(recwarn) == 1 assert recwarn.pop(DeprecationWarning) def test_type(recwarn): warnings.simplefilter("always") ie = IECore() net = ie.read_network(model=test_net_xml, weights=test_net_bin) assert net.layers['27'].type == "Pooling" assert len(recwarn) == 1 assert recwarn.pop(DeprecationWarning) def test_affinity_getter(recwarn): warnings.simplefilter("always") ie = IECore()
# Copyright (C) 2018-2021 Intel Corporation # SPDX-License-Identifier: Apache-2.0 import numpy as np import os import pytest import warnings import threading from datetime import datetime import time from openvino.inference_engine import ie_api as ie from conftest import model_path, image_path is_myriad = os.environ.get("TEST_DEVICE") == "MYRIAD" test_net_xml, test_net_bin = model_path(is_myriad) path_to_img = image_path() def create_function_with_memory(input_shape, data_type): from ngraph.impl import Function, Type import ngraph as ng input_data = ng.parameter(input_shape, name="input_data", dtype=data_type) rv = ng.read_value(input_data, "var_id_667") add = ng.add(rv, input_data, name="MemoryAdd") node = ng.assign(add, "var_id_667") res = ng.result(add, "res") func = Function(results=[res], sinks=[node], parameters=[input_data], name="name")
def test_read_network_as_path(): ie = IECore() net = ie.read_network(model=Path(model_path()[0]), weights=Path(test_net_bin)) assert isinstance(net, IENetwork)