ATC yolov7 不能 AIPP 使能是为什么
收藏回复举报
ATC yolov7 不能 AIPP 使能是为什么
t('forum.solved') 已解决
发表于2023-06-30 10:07:00
0 查看

模型转换成功后,查看模型结构时却没有 aipp 的结点,是什么原因呢?转换 yolov5 模型,并使用相同的 aipp 配置文件,查看模型结构是有 aipp 结点。这是为什么呢?

模型转换的命令行

atc --model=yolov7.onnx \
    --insert_op_conf=/data/work/ascend/data/convert/yolov5_c/aipp.cfg \
    --framework=5 \
    --output=yolov7 \
    --soc_version=Ascend310 \
    --log=error

aipp.cfg 配置如下

aipp_op {
    aipp_mode: static
    input_format: YUV420SP_U8
    related_input_rank: 0
    # related_input_name: "images"
    csc_switch: true
    rbuv_swap_switch: false
    src_image_size_w: 416
    src_image_size_h: 416
    matrix_r0c0: 298
    matrix_r0c1: 0
    matrix_r0c2: 409
    matrix_r1c0: 298
    matrix_r1c1: -100
    matrix_r1c2: -208
    matrix_r2c0: 298
    matrix_r2c1: 516
    matrix_r2c2: 0
    input_bias_0: 16
    input_bias_1: 128
    input_bias_2: 128
}

我要发帖子