example-apps/PDD/pcb-defect-detection/libs/label_name_dict/remote_sensing_dict.py

   1 # -*- coding: utf-8 -*-
   2
   3 NAME_LABEL_MAP = {
   4     'back_ground': 0,
   5     'building': 1
   6 }
   7
   8
   9 def get_label_name_map():
  10     reverse_dict = {}
  11     for name, label in NAME_LABEL_MAP.items():
  12         reverse_dict[label] = name
  13     return reverse_dict
  14
  15 LABEl_NAME_MAP = get_label_name_map()