diff --git a/tool/paddle2nb.py b/tool/paddle2nb.py new file mode 100644 index 0000000..9414cf1 --- /dev/null +++ b/tool/paddle2nb.py @@ -0,0 +1,16 @@ +from paddlelite.lite import Opt + +if __name__ == '__main__': + # 1. 创建opt实例 + opt = Opt() + # 2. 指定输入模型地址 + opt.set_model_dir("./model") + # 3. 指定转化类型: arm、x86、opencl、npu + # 一般认为arm为cpu,opencl为gpu + opt.set_valid_places("arm") + # 4. 指定模型转化类型: naive_buffer、protobuf + opt.set_model_type("naive_buffer") + # 4. 输出模型地址 + opt.set_optimize_out("model") + # 5. 执行模型优化 + opt.run()