1 files changed, 28 insertions, 0 deletions
diff --git a/tools/nnpackage_tool/tflite2circle/tflitejson2circlejson.py b/tools/nnpackage_tool/tflite2circle/tflitejson2circlejson.py
new file mode 100644
index 000000000..c20a0c53e
--- /dev/null
+++ b/tools/nnpackage_tool/tflite2circle/tflitejson2circlejson.py
@@ -0,0 +1,28 @@
+#!/usr/bin/python3
+
+import json
+import os
+import sys
+from collections import OrderedDict
+
+
+def usage():
+    script = os.path.basename(os.path.basename(__file__))
+    print("Usage: {} path_to_tflite_in_json".format(script))
+    sys.exit(-1)
+
+
+if __name__ == '__main__':
+    if len(sys.argv) != 2:
+        usage()
+
+    json_path = sys.argv[1]
+    with open(json_path, "r") as f:
+        try:
+            json_dict = json.load(f, object_pairs_hook=OrderedDict)
+            for subgraph in json_dict["subgraphs"]:
+                subgraph["data_format"] = "CHANNELS_LAST"
+            print(json.dumps(json_dict, indent=2))
+        except KeyError:
+            print("subgraphs attribute does not exist.")
+            sys.exit(-2)