提交 b38d642a 编写于 作者: S syyxsxx

update deploy.py

上级 ff4f0bac
...@@ -65,9 +65,9 @@ class Predictor: ...@@ -65,9 +65,9 @@ class Predictor:
logging.info("Creating Inference Engine") logging.info("Creating Inference Engine")
ie = IECore() ie = IECore()
logging.info("Loading network files:\n\t{}\n\t{}".format(self.model_xml, self.model_bin)) logging.info("Loading network files:\n\t{}\n\t{}".format(self.model_xml, self.model_bin))
net = ie.read_network(model=self.model_xml,weights=self.model_bin) net = ie.read_network(model=self.model_xml, weights=self.model_bin)
net.batch_size = 1 net.batch_size = 1
exec_net = ie.load_network(network=net,device_name=self.device) exec_net = ie.load_network(network=net, device_name=self.device)
return exec_net, net return exec_net, net
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册