pcb defect detetcion application

[ealt-edge.git] / example-apps / PDD / pcb-defect-detection / libs / label_name_dict / remote_sensing_dict.py

diff --git a/example-apps/PDD/pcb-defect-detection/libs/label_name_dict/remote_sensing_dict.py b/example-apps/PDD/pcb-defect-detection/libs/label_name_dict/remote_sensing_dict.py
new file mode 100755 (executable)
index 0000000..5257bc9
--- /dev/null
+++ b/example-apps/PDD/pcb-defect-detection/libs/label_name_dict/remote_sensing_dict.py
@@ -0,0 +1,15 @@
+# -*- coding: utf-8 -*-
+
+NAME_LABEL_MAP = {
+    'back_ground': 0,
+    'building': 1
+}
+
+
+def get_label_name_map():
+    reverse_dict = {}
+    for name, label in NAME_LABEL_MAP.items():
+        reverse_dict[label] = name
+    return reverse_dict
+
+LABEl_NAME_MAP = get_label_name_map()
\ No newline at end of file