加了AIPP转om模型后精度异常,onnx和不加AIPP的om是正常推理的
收藏回复举报
加了AIPP转om模型后精度异常,onnx和不加AIPP的om是正常推理的
t('forum.solved') 已解决
新人帖
发表于2025-04-24 10:43:09
0 查看

onnx正常推理结果

cke_7626.png

带AIPP的om模型推理结果:

cke_6076.png

输出的置信度都非常低是为什么

我的AIPP配置:aipp_op { aipp_mode: static input_format: RGB888_U8 src_image_size_w: 1280 src_image_size_h: 720 csc_switch: false crop: true load_start_pos_h: 0 load_start_pos_w: 0 crop_size_w: 640 crop_size_h: 640 var_reci_chn_0: 0.00392156862745 var_reci_chn_1: 0.00392156862745 var_reci_chn_2: 0.00392156862745 } 

我的ATC指令: 

#!/bin/bash 

 

# ======== 配置参数 ======== 

ONNX_MODEL="./yolov8s.onnx" 

OUTPUT_MODEL="./yolov8s_aipp_RGB_center_crop" 

AIPP_CFG="./aipp_RGB.cfg" 

SOC_VERSION="Ascend310B4" 

 

# ======== 执行 ATC 转换 ======== 

echo "开始将 $ONNX_MODEL 转换为带 AIPP 的 OM 模型..." 

 

atc \ 

  --model=${ONNX_MODEL} \ 

  --output=${OUTPUT_MODEL} \ 

  --framework=5 \ 

  --input_shape="images:1,3,640,640" \ 

  --insert_op_conf=${AIPP_CFG} \ 

  --soc_version=${SOC_VERSION} \ 

  --precision_mode=allow_fp32_to_fp16 

  # --log=debug 

 

我要发帖子