paddlelite_export.py 665 B

12345678910111213141516171819
  1. # 需先安装paddlelite
  2. import paddlelite.lite as lite
  3. model_filename = 'server_mobilenet/__model__'
  4. params_filename = 'server_mobilenet/__params__'
  5. export_filename = 'mobilenetv2'
  6. opt = lite.Opt()
  7. # 将正常模型导出为Lite模型
  8. opt.run_optimize("", model_filename, params_filename, 'naive_buffer', 'arm',
  9. export_filename)
  10. quant_model_filename = 'quant_mobilenet/__model__'
  11. quant_params_filename = 'quant_mobilenet/__params__'
  12. quant_export_filename = 'mobilenetv2_quant'
  13. # 将量化模型导出为Lite模型
  14. opt.run_optimize("", quant_model_filename, quant_params_filename,
  15. 'naive_buffer', 'arm', quant_export_filename)