diff --git a/configs/picodet/README.md b/configs/picodet/README.md index 99cfb55e3c8e717d0469690b1c87f1e386b593f7..b6562428308b83a4aa49570a7a983a081beace7c 100644 --- a/configs/picodet/README.md +++ b/configs/picodet/README.md @@ -194,18 +194,13 @@ paddle2onnx --model_dir output_inference/picodet_s_320_coco_lcnet/ \ - 简化ONNX模型: 使用`onnx-simplifier`库来简化ONNX模型。 - - 安装 onnx-simplifier >= 0.3.6: + - 安装 onnxsim >= 0.4.1: ```shell - pip install onnx-simplifier + pip install onnxsim ``` - 简化ONNX模型: ```shell - python -m onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx - ``` - - 如果模型包含所有后处理，简化模型时需要指定`dynamic-input-shape`： - ```shell - python -m onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx --dynamic-input-shape --input-shape image:1,3,320,320 + onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx ``` diff --git a/configs/picodet/README_en.md b/configs/picodet/README_en.md index da2e1dfc288d4a10e1e1d3d25c7b61dacfafd23b..d7d51c7b3b774ce0f68822a7d5084ea8639ada53 100644 --- a/configs/picodet/README_en.md +++ b/configs/picodet/README_en.md @@ -189,18 +189,13 @@ paddle2onnx --model_dir output_inference/picodet_s_320_coco_lcnet/ \ - Simplify ONNX model: use onnx-simplifier to simplify onnx model. - - Install onnx-simplifier >= 0.3.6: + - Install onnxsim >= 0.4.1: ```shell - pip install onnx-simplifier + pip install onnxsim ``` - simplify onnx model: ```shell - python -m onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx - ``` - - If the model includes postprocessing, specify `dynamic-input-shape` when simplifying the model: - ```shell - python -m onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx --dynamic-input-shape --input-shape image:1,3,320,320 + onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx ```