Temples-Classifier/constants.py at master · mhashas/Temples-Classifier · GitHub

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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# Model constants
PSPNET = 'pspnet'
PSPNET_50 = 'pspnet_50'
VGG_19 = 'vgg_19'
VGG_16 = 'vgg'
RESNET = 'resnet'
RESNET_50 = 'resnet_50'

# Loss functions
CE_LOSS = 'ce_loss'

# Dataset
TEMPLES_DATASET = 'temples_dataset'

# Optimizers
SGD = 'sgd'
AMSGRAD = 'amsgrad'
ADAM = 'adam'
RMSPROP = 'rmsprop'
ADABOUND = 'adabound'

# Normalization layers
INSTANCE_NORM = 'instance'
BATCH_NORM = 'batch'
SYNC_BATCH_NORM = 'syncbn'

# Init types
NORMAL_INIT = 'normal'
KAIMING_INIT = 'kaiming'
XAVIER_INIT = 'xavier'
ORTHOGONAL_INIT = 'orthogonal'

# Downsampling methods
MAXPOOL = 'maxpool'
STRIDECONV = 'strideconv'

# Split constants
TRAIN = 'train'
VAL = 'val'
TEST = 'test'
TRAINVAL = 'trainval'
VISUALIZATION = 'visualization'