-
Notifications
You must be signed in to change notification settings - Fork 0
/
constants.py
37 lines (33 loc) · 840 Bytes
/
constants.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
26
27
28
29
30
31
32
33
34
35
36
37
INPUT_WIDTH = 224
INPUT_HEIGHT = INPUT_WIDTH
FORMAT_LEADING_ZEROS = ':03d'
TFRECORDS_FORMAT_PATTERN = '{}-{' + FORMAT_LEADING_ZEROS + '}-of-{' + FORMAT_LEADING_ZEROS + '}.tfrecords'
CLASS_NO = 21 # background + classes, without void
DATASET_DIRECTORY_PATH = "VOCdevkit/VOC2012"
TFRECORDS_SAVE_PATH = "tfrecords"
LABEL_IDS_TO_NAME = {
0: "background",
1: "aeroplane",
2: "bicycle",
3: "bird",
4: "boat",
5: "bottle",
6: "bus",
7: "car",
8: "cat",
9: "chair",
10: "cow",
11: "diningtable",
12: "dog",
13: "horse",
14: "motorbike",
15: "person",
16: "potted plant",
17: "sheep",
18: "sofa",
19: "train",
20: "tv/monitor",
255: "void"
}
LABEL_NAMES_TO_ID = {value: key for key, value in LABEL_IDS_TO_NAME.items()}
VGG_MEAN = [103.939, 116.779, 123.68]