本章节以TensorFlow框架ResNet50网络模型为例进行演示,参见1获取原始模型文件。
atc --model=$HOME/module/resnet50_tensorflow*.pb --framework=3 --output=$HOME/module/out/tf_resnet50 --soc_version=<soc_version> --input_shape="Placeholder:-1,-1,-1,3" --dynamic_dims="1,224,224;8,448,448" --input_format=ND
成功执行命令后,在--output参数指定的路径下,可查看离线模型。
模型转换完成后,在生成的om模型中,会新增一个输入,在模型推理时通过该新增的输入提供具体的维度值。例如,a输入的维度为动态的,在om模型中,会有与a对应的b输入来描述a的维度值。