Fix wrong api in fastdeploy model

2025-10-05 00:33:03 +08:00 · 2022-11-22 06:25:05 +00:00
parent 1a9a2df782
commit 5d37f739e8
2 changed files with 14 additions and 12 deletions
--- a/examples/vision/detection/paddledetection/python/infer_ppyoloe.py
+++ b/examples/vision/detection/paddledetection/python/infer_ppyoloe.py
@@ -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: