Browse Source

add manuscript for exporting lite model

FlyingQianMM 5 năm trước cách đây
mục cha
commit
395b35dcb3
1 tập tin đã thay đổi với 34 bổ sung0 xóa
  1. 34 0
      deploy/lite/export_lite.py

+ 34 - 0
deploy/lite/export_lite.py

@@ -0,0 +1,34 @@
+import paddlelite.lite as lite
+import os
+import argparse
+
+
+def export_lite():
+    opt = lite.Opt()
+    model_file = os.path.join(FLAGS.model_path, '__model__')
+    params_file = os.path.join(FLAGS.model_path, '__params__')
+    opt.run_optimize("", model_file, params_file, FLAGS.place, FLAGS.save_dir)
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument(
+        "--model_path",
+        type=str,
+        default="",
+        help="model path.",
+        required=True)
+    parser.add_argument(
+        "--place",
+        type=str,
+        default="arm",
+        help="preprocess config path.",
+        required=True)
+    parser.add_argument(
+        "--save_dir",
+        type=str,
+        default="paddlex.onnx",
+        help="Directory for storing the output visualization files.",
+        required=True)
+    FLAGS = parser.parse_args()
+    export_lite()