以下介绍通过ATC模型转换工具获取离线模型的操作步骤,更多操作请参见《ATC工具使用指南》。
例如:resnet50.prototxt和resnet50.caffemodel
atc --model=$HOME/module/resnet50.prototxt --weight=$HOME/module/resnet50.caffemodel --framework=0 --output=$HOME/module/out/caffe_resnet50 --soc_version=<soc_version>
ATC run success
成功执行命令后,在--output参数指定的路径下,可查看离线模型(如:resnet50.om)。
atc --mode=1 --om=$HOME/module/out/caffe_resnet50/resnet50.om --json=$HOME/data/resnet50.json
若提示如下信息,则说明转换json文件成功。
ATC run success
成功执行命令后,在--json参数指定的路径下可查看转换后的json文件。
请参见《AMCT工具(Caffe)》中的“快速入门”获取量化原始模型和量化信息文件。
resnet50_deploy_model.prototxt和resnet50_deploy_weights.caffemodel文件可用于进行ATC模型转换,resnet50_fake_quant_model.prototxt和resnet50_fake_quant_weights.caffemodel可用于进行Caffe量化原始模型的dump操作。
将从量化原始模型和量化信息文件操作中获取到的量化原始模型文件resnet50_deploy_model.prototxt和resnet50_deploy_weights.caffemodel,以全网层信息文件中的ATC操作执行模型转换即可获取到量化离线模型文件和量化离线模型文件转换的json文件。