From 08c2168b810bd2840e29729a212f563e90fa6224 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=83=AD=E5=AE=A3?= <2698345810@qq.com> Date: Mon, 5 Sep 2022 07:04:50 +0000 Subject: [PATCH] update YOLOv5Prune/README.md. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 郭宣 <2698345810@qq.com> --- YOLOv5Prune/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/YOLOv5Prune/README.md b/YOLOv5Prune/README.md index d1084fc..3c087d0 100644 --- a/YOLOv5Prune/README.md +++ b/YOLOv5Prune/README.md @@ -190,7 +190,7 @@ ONNX模型可通过[地址](https://mindx.sdk.obs.cn-north-4.myhuaweicloud.com/a 2. 执行命令： ``` -atc --model=prune55_t.onnx --framework=5 --output=./prune60_t --input_format=NCHW --input_shape="images:1,3,512,512" --enable_small_channel=1 --insert_op_conf=./aipp_yolov5.cfg --soc_version=Ascend310 --out_nodes="Transpose_260:0;Transpose_520:0;Transpose_780:0" #a +atc --model=prune55_t.onnx --framework=5 --output=./prune55_t --input_format=NCHW --input_shape="images:1,3,512,512" --enable_small_channel=1 --insert_op_conf=./aipp_yolov5.cfg --soc_version=Ascend310 --out_nodes="Transpose_260:0;Transpose_520:0;Transpose_780:0" #a #--out_nodes="Transpose_260:0;Transpose_556:0;Transpose_825:0" #b #可通过netron工具查看out_nodes,不同权重可能会存在差异。 #如果使用本仓提供的权重，a对应prune55和prune60，b对应prune50。