Browse Source

update deploy.py

syyxsxx 5 years ago
parent
commit
b38d642a82
1 changed files with 2 additions and 2 deletions
  1. 2 2
      deploy/openvino/python/deploy.py

+ 2 - 2
deploy/openvino/python/deploy.py

@@ -65,9 +65,9 @@ class Predictor:
         logging.info("Creating Inference Engine")
         ie = IECore()
         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
-        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