fix some usage problem in linux (#25)

Co-authored-by: root <root@bjyz-sys-gpu-kongming3.bjyz.baidu.com>
This commit is contained in:
Jason
2022-07-19 11:47:13 +08:00
committed by GitHub
parent e7c6a9d346
commit f670520bf8
5 changed files with 56 additions and 17 deletions

View File

@@ -53,3 +53,29 @@ class FastDeployModel:
if self._model is None:
return False
return self._model.initialized()
class FastDeployRuntime:
def __init__(self, runtime_option):
self._runtime = C.Runtime();
assert self._runtime.init(runtime_option), "Initialize FastDeployRuntime Failed!"
def infer(self, data):
assert isinstance(data, dict), "The input data should be type of dict."
return self._runtime.infer(data)
def num_inputs(self):
return self._runtime.num_inputs();
def num_outputs(self):
return self._runtime.num_outputs();
def get_input_info(self, index):
assert isinstance(index, int), "The input parameter index should be type of int."
assert index < self.num_inputs(), "The input parameter index:{} should less than number of inputs:{}.".format(index, self.num_inputs)
return self._runtime.get_input_info(index)
def get_output_info(self, index):
assert isinstance(index, int), "The input parameter index should be type of int."
assert index < self.num_outputs(), "The input parameter index:{} should less than number of outputs:{}.".format(index, self.num_outputs)
return self._runtime.get_output_info(index)