CANN训练:模型推理时数据预处理方法及归一化参数计算

💡 原文中文,约2200字,阅读约需6分钟。
📝

内容提要

本文比较了基于Ascend CL模型推理时使用的OpenCV、AIPP和DVPP三种方式的特点,并以Resnet50的pytorch模型为例,说明了它们如何实现预处理。文章还介绍了AIPP计算减均值和乘系数的参数。

🎯

关键要点

  • 本文比较了基于Ascend CL模型推理时使用的OpenCV、AIPP和DVPP三种方式的特点。
  • 以Resnet50的pytorch模型为例,说明了如何实现预处理。
  • 推理前需要对图片进行预处理,以满足模型的输入要求。
  • Resnet50模型的预处理方法包括缩放、RGB顺序存放、归一化和减均值乘系数。
  • 查找模型训练时的预处理方式可以通过Ascend ModelZoo或PyTorch官网。
  • Ascend CL常用的预处理方式有三种:OpenCV、AIPP、DVPP。
  • AIPP的减均值和乘系数的计算公式提供了具体的数学表达。
  • 均值和标准差的参数分别为:[0.485, 0.456, 0.406]和[0.229, 0.224, 0.225]。
➡️

继续阅读