paddlelite_export.py 634 B

123456789101112131415161718
  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', export_filename)
  9. quant_model_filename = 'quant_mobilenet/__model__'
  10. quant_params_filename = 'quant_mobilenet/__params__'
  11. quant_export_filename = 'mobilenetv2_quant'
  12. # 将量化模型导出为Lite模型
  13. opt.run_optimize("", quant_model_filename, quant_params_filename, 'naive_buffer', 'arm', quant_export_filename)