1 # -*- coding: utf-8 -*-
2 from __future__ import division, print_function, absolute_import
4 from libs.configs import cfgs
6 if cfgs.DATASET_NAME == 'ship':
11 elif cfgs.DATASET_NAME == 'aeroplane':
16 elif cfgs.DATASET_NAME == 'WIDER':
21 elif cfgs.DATASET_NAME == 'jyzdata':
30 elif cfgs.DATASET_NAME == 'icdar':
35 elif cfgs.DATASET_NAME.startswith('DOTA'):
42 'soccer-ball-field': 5,
49 'ground-track-field': 12,
51 'basketball-court': 14,
52 'baseball-diamond': 15
54 elif cfgs.DATASET_NAME == 'pascal':
78 elif cfgs.DATASET_NAME == 'pcb':
89 assert 'please set label dict!'
92 def get_label_name_map():
94 for name, label in NAME_LABEL_MAP.items():
95 reverse_dict[label] = name
98 LABEl_NAME_MAP = get_label_name_map()