From fd5ea88e13de0e3a1070ae82912bef2632fc8aa5 Mon Sep 17 00:00:00 2001 From: bruce1408 Date: Thu, 21 Mar 2024 15:35:10 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=AA=20test(onnxsim):=20=E4=BF=AE?= =?UTF-8?q?=E6=94=B9onnxsim=E5=87=BD=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Tools/onnx_model_convert_onnxSimplify.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Tools/onnx_model_convert_onnxSimplify.py b/Tools/onnx_model_convert_onnxSimplify.py index 957e9d9..96b4098 100644 --- a/Tools/onnx_model_convert_onnxSimplify.py +++ b/Tools/onnx_model_convert_onnxSimplify.py @@ -55,7 +55,8 @@ def print_onnx_input_output(model_path): print("Shape:", shape, "Type:", onnx.TensorProto.DataType.Name(data_type)) if __name__=="__main__": - model_path = "/Users/bruce/Downloads/8620_deploy/swin_tiny_patch4_window7_224_224_elementwise_affine.onnx" + # model_path = "/Users/bruce/Downloads/8620_deploy/swin_tiny_patch4_window7_224_224_elementwise_affine.onnx" + model_path = "/mnt/share_disk/bruce_cui/onnx_models/laneline_154w_20240320_fastbev.onnx" # model_path = "/Users/bruce/Downloads/8620_deploy/Laneline/models/epoch_latest_0302.onnx" onnx_simplify(model_path)