下载
中文
注册

AIPP对模型输入大小的校验说明

如果有配置AIPP,无论静态AIPP还是动态AIPP,最终生成离线模型的输入大小(即input_size)均会被Crop、Padding等操作影响。本节给出对模型输入大小的约束说

假设模型的Batch数量为N(如果为动态batch场景,N为最大档位数的取值),模型输入图片的宽为src_image_size_w,高为src_image_size_h,最后模型输入的Size的计算公式如下所示。

静态AIPP对模型输入大小的校验

表1 input_size校验公式

input_format

input_size

备注

YUV400_U8

N * src_image_size_w * src_image_size_h * 1

-

YUV420SP_U8

N * src_image_size_w * src_image_size_h * 1.5

-

XRGB8888_U8

N * src_image_size_w * src_image_size_h * 4

-

RGB888_U8

N * src_image_size_w * src_image_size_h * 3

-

动态AIPP对模型输入大小的校验

如果为动态AIPP,模型转换时,ATC会为动态AIPP新增一个模型输入,用于接收模型推理阶段通过调用aclmdlSetInputAIPP接口后传入的AIPP参数,该场景下新增输入节点大小计算公式为:

sizeof(kAippDynamicPara) - sizeof(kAippDynamicBatchPara) + batch_count * sizeof(kAippDynamicBatchPara)

kAippDynamicPara以及kAippDynamicBatchPara参数解释请参见动态AIPP的参数输入结构