-
Notifications
You must be signed in to change notification settings - Fork 2
/
const.py
25 lines (24 loc) · 661 Bytes
/
const.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
# Chest x-ray 14
CATEGORIES_MAP = {
'No Finding' : 0,
'Infiltration' : 1,
'Atelectasis' : 2,
'Effusion' : 3,
'Nodule' : 4,
'Pneumothorax' : 5,
'Mass' : 6,
'Consolidation' : 7,
'Pleural_Thickening' : 8,
'Cardiomegaly' : 9,
'Emphysema' : 10,
'Fibrosis' : 11,
'Edema' : 12,
'Pneumonia' : 13,
'Hernia' : 14,
}
INVERT_CATEGORIES_MAP = ['No Finding', 'Infiltration', 'Atelectasis',
'Effusion', 'Nodule', 'Pneumothorax', 'Mass',
'Consolidation', 'Pleural_Thickening', 'Cardiomegaly',
'Emphysema', 'Fibrosis', 'Edema', 'Pneumonia','Hernia']
# Project dir
BASE_DIR = '/content/drive/My Drive/GAN'