文档首页> Atlas 500应用> 模型转换> AIPP配置> AIPP对模型输入大小的校验说明
更新时间:2021-03-18 GMT+08:00
分享

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

如果有配置AIPP,无论静态AIPP还是动态AIPP,最终生成的Davinci模型的接收的图片的大小(即input_size)均会被Crop、Padding等操作影响。在模型推理阶段,OME会对传入的图片的大小进行校验,如果是动态AIPP,要求模型推理时传入的图片的大小小于等于动态AIPP配置的max_src_image_size大小;如果是静态AIPP,要求模型推理时传入的图片的大小与表1中的计算的input_size相等。

对于静态AIPP,假设模型的Batch数量为N,图片的宽为src_image_size_w,高为src_image_size_h,最后模型输入的Size的计算公式如表1所示。

表1 input_size校验公式

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)

分享:

    相关文档

    相关产品