AIPP对模型输入大小的校验说明
如果有配置AIPP,无论静态AIPP还是动态AIPP,最终生成离线模型的输入大小(即input_size)均会被Crop、Padding等操作影响。本节给出对模型输入大小的约束说明。
假设模型的Batch数量为N(如果为动态batch场景,N为最大档位数的取值),模型输入图片的宽为src_image_size_w,高为src_image_size_h,最后模型输入的Size的计算公式如下所示。
静态AIPP对模型输入大小的校验
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的参数输入结构。
父主题: AIPP使能