Created by: DannyIsFunny
【背景】: 有需求需要将模型优化为2.1.0之前版本的格式,即优化模型为两个文件 【本PR工作】:将模型保存为v2.1.0之前版本样式 【todo】: 将增加opt选项,使opt 可以选择优化模型为老格式或者新格式(合入develop分支) 【warning】: 相关单测没有修改,可能CI无法测试通过。 opt 功能验证正常,输出符合预期
【验证方法】
1、编译opt
./lite/tools/build.sh build_optimize_tool
2、使用opt转化模型
./opt \
--model_dir=./mobilenet_v1 \
--optimize_out_type=naive_buffer \
--optimize_out=mobilenetv1_opt \
--valid_targets=arm
输出结果
- mobilenetv1_opt
|-- params.nb
|--model.nb
3、使用CXX_DEMO验证模型是否正常
# 编译demo,将mobile_light_api.cc 中的 set_model_from_file修改为set_model_dir()
cd inference_lite_lib/demo/cxx/mobile_light
make
在手机上测试:
adb push ********
adb shell 'cd /data/local/tmp && export LD_LIBRARY_PATH=./ && ./mobile_light_api ./mobilenetv1_opt'