# cv_models = {'wbc_1024x1024':['phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-1',
# 'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-2',
# 'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-3',
# 'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-4',
# 'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-5',
# 'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-1',
# 'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-2',
# 'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-3',
# 'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-4',
# 'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-5'],
# # wbc2_1024x1024
# 'wbc2_1024x1024':['phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-1',
# 'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-2',
# 'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-3',
# 'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-4',
# 'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-5',
# 'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-1',
# 'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-2',
# 'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-3',
# 'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-4',
# 'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-5'],
#  # wbc4_1024x1024
# 'wbc4_1024x1024':['phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-1',
# 'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-2',
# 'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-3',
# 'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-4',
# 'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-4-valid-False-cross-5',
# 'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-1',
# 'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-2',
# 'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-3',
# 'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-4',
# 'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-8-snap-imagenet-cls-5-valid-False-cross-5']}

# Mar 21, 2021
# cv_models = {'wbc_1024x1024':['phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-1',
# 'phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-2',
# 'phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-3',
# 'phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-4',
# 'phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-5',
# 'phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-1',
# 'phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-2',
# 'phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-3',
# 'phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-4',
# 'phi-1-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-5'],
# # wbc2_1024x1024
# 'wbc2_1024x1024':['phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-1',
# 'phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-2',
# 'phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-3',
# 'phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-4',
# 'phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-5',
# 'phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-1',
# 'phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-2',
# 'phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-3',
# 'phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-4',
# 'phi-1-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-5'],
#  # wbc4_1024x1024
# 'wbc4_1024x1024':['phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-1',
# 'phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-2',
# 'phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-3',
# 'phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-4',
# 'phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-4-valid-False-cross-5',
# 'phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-1',
# 'phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-2',
# 'phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-3',
# 'phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-4',
# 'phi-1-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-4-snap-imagenet-cls-5-valid-False-cross-5']}

# Mar 22, 2021
cv_models = {'wbc_1024x1024':['phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-1',
'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-2',
'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-3',
'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-4',
'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-5',
'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-1',
'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-2',
'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-3',
'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-4',
'phi-0-set-wbc_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-5'],
# wbc2_1024x1024
'wbc2_1024x1024':['phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-1',
'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-2',
'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-16-snap-imagenet-cls-4-valid-False-cross-3',
'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-16-snap-imagenet-cls-4-valid-False-cross-4',
'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-5',
'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-1',
'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-2',
'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-3',
'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-4',
'phi-0-set-wbc2_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-5'],
 # wbc4_1024x1024
'wbc4_1024x1024':['phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-1',
'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-2',
'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-3',
'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-4',
'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-4-valid-False-cross-5',
'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-1',
'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-2',
'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-3',
'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-4',
'phi-0-set-wbc4_1024x1024-wfpn-False-ep-200-stp-100-bz-12-snap-imagenet-cls-5-valid-False-cross-5']}

cls4_label_map = {'neutrophils': 0, 'eosinophils': 1, 'lymphocytes': 2, 'monocytes': 3}
cls5_label_map = {'neutrophils': 0, 'bands': 1, 'eosinophils': 2, 'lymphocytes': 3, 'monocytes': 4}

label_cls4_map = {0:'neutrophils', 1:'eosinophils', 2:'lymphocytes', 3:'monocytes'}
label_cls5_map = {0:'neutrophils', 1:'bands', 2:'eosinophils', 3:'lymphocytes', 4:'monocytes'}