apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix wrong api in fastdeploy model

Browse Source
This commit is contained in:
jiangjiajun
2022-11-22 06:25:05 +00:00
parent 1a9a2df782
commit 5d37f739e8
2 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/vision/detection/paddledetection/python/infer_ppyoloe.py
View File

@@ -53,6 +53,8 @@ config_file = os.path.join(model_dir, "infer_cfg.yml")
runtime_option = build_option(args)
model = fd.vision.detection.PPYOLOE(
model_file, params_file, config_file, runtime_option=runtime_option)
print(model._model.input_info_of_runtime(0))
print(model._model.output_info_of_runtime(0))
# 预测图片检测结果
if args.image is None:

24
python/fastdeploy/model.py
View File

@@ -27,23 +27,23 @@ class FastDeployModel:
def model_name(self):
return self._model.model_name()
def num_inputs(self):
return self._model.num_inputs()
def num_inputs_of_runtime(self):
return self._model.num_inputs_of_runtime()
def num_outputs(self):
return self._model.num_outputs()
def num_outputs_of_runtime(self):
return self._model.num_outputs_of_runtime()
def get_input_info(self, index):
assert index < self.num_inputs(
def input_info_of_runtime(self, index):
assert index < self.num_inputs_of_runtime(
), "The index:{} must be less than number of inputs:{}.".format(
index, self.num_inputs())
return self._model.get_input_info(index)
index, self.num_inputs_of_runtime())
return self._model.input_info_of_runtime(index)
def get_output_info(self, index):
assert index < self.num_outputs(
def output_info_of_runtime(self, index):
assert index < self.num_outputs_of_runtime(
), "The index:{} must be less than number of outputs:{}.".format(
index, self.num_outputs())
return self._model.get_output_info(index)
index, self.num_outputs_of_runtime())
return self._model.output_info_of_runtime(index)
def enable_record_time_of_runtime(self):
self._model.enable_record_time_of_runtime()