AIPP对模型输入大小的校验说明
如果有配置AIPP,无论静态AIPP还是动态AIPP,最终生成的Davinci模型的接收的图片的大小(即input_size)均会被Crop、Padding等操作影响。在模型推理阶段,OME会对传入的图片的大小进行校验,如果是动态AIPP,要求模型推理时传入的图片的大小小于等于动态AIPP配置的max_src_image_size大小;如果是静态AIPP,要求模型推理时传入的图片的大小与表11-7中的计算的input_size相等。
对于静态AIPP,假设模型的Batch数量为N,图片的宽为src_image_size_w,高为src_image_size_h,最后模型输入的Size的计算公式如表11-7所示。
input_format |
input_size |
---|---|
YUV400_U8 |
N * src_image_size_w * src_image_size_h |
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的模型时,OMG会为动态AIPP添加模型输入。新增动态AIPP输入的input_size计算公式为:
sizeof(kAippDynamicPara) + (batch_count - 1) * sizeof(kAippDynamicBatchPara)