X-Git-Url: https://gerrit.akraino.org/r/gitweb?a=blobdiff_plain;f=example-apps%2FPDD%2Fpcb-defect-detection%2Flibs%2Flabel_name_dict%2Fremote_sensing_dict.py;fp=example-apps%2FPDD%2Fpcb-defect-detection%2Flibs%2Flabel_name_dict%2Fremote_sensing_dict.py;h=5257bc9cd55663110a4a468960ca9f55f1917221;hb=a785567fb9acfc68536767d20f60ba917ae85aa1;hp=0000000000000000000000000000000000000000;hpb=94a133e696b9b2a7f73544462c2714986fa7ab4a;p=ealt-edge.git

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
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