From: 최성진/동작제어Lab(SR)/Principal Engineer/삼성전자 <lotieye.choi@samsung.com>
Date: Wed, 16 May 2018 02:49:55 +0000 (+0900)
Subject: Add CustomOptions for supporting ADAS models (#1227)
X-Git-Tag: 0.2~905
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=1f11aaa548d38bee12a35d198a9e0f121bc41a59;p=platform%2Fcore%2Fml%2Fnnfw.git

Add CustomOptions for supporting ADAS models (#1227)

This commit introduces CustomOptions to support ADAS model in print_op tool file.

- add custom_options as well as builtin_options_type

Signed-off-by: SungJin Choi <lotieye.choi@samsung.com>
---

diff --git a/tools/extract_weights_from_tflite/print_op.py b/tools/extract_weights_from_tflite/print_op.py
index 69962bf..1db3c22 100755
--- a/tools/extract_weights_from_tflite/print_op.py
+++ b/tools/extract_weights_from_tflite/print_op.py
@@ -37,7 +37,13 @@ tensors = j['subgraphs'][0]['tensors']
 operators = j['subgraphs'][0]['operators']
 
 for o in operators:
-    print "Layer: ", o['builtin_options_type']
+    if o.has_key("builtin_options_type"):
+        print "Layer: ", o['builtin_options_type']
+    elif o.has_key("custom_options"):
+        print "Layer: CustomOptions"
+    else:
+        print "Layer: NotSupportedOptions"
+        continue
     print "    Input shapes ---"
     for inp in o['inputs']:
         print "      ", tensors[inp]['shape']