实现了MobileNetV2的推理

应用分析

下载的模型为MobileNetv2，是一个图像分类模型。整体的开发流程为资源初始化->DVPP初始化和模型加载->数据预处理->模型推理->数据后处理->资源释放

输入输出数据

输入数据为224*224大小的图片，格式为NHWC（TF模型转换过来的）；
输出为一千维的向量；

数据预处理和后处理

预处理需要色域转换、尺寸缩放、数据归一化；对于输出数据选择置信度最高的五个输出索引值。

模型转换

ATC转换代码

1
atc --model=mobileNetv2.pb --framework=3 --output=mobilenet_v2_aipp --output_type=FP32 --soc_version=Ascend310 --input_shape="input:1,224,224,3" --log=info --insert_op_conf=mobilenet_v2_tf_aipp.cfg
  • framework=3指TensorFlow；
  • input_shape为 NHWC；
  • insert_op_conf中配置了色域转换和数据归一化。

运行结果