CS231A_project/config.py at master · Flyhigh2017/CS231A_project · 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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
import os

#
# path and dataset parameter
#

DATA_PATH = '/Users/anekisei/Documents/cs231a_project/data/data_tracking_image_2/training/image_02'

LABEL_PATH = '/Users/anekisei/Documents/cs231a_project/data/label_02'


OUTPUT_DIR = os.path.join(DATA_PATH, 'output1')

WEIGHTS_DIR = os.path.join(DATA_PATH, 'weights1')

WEIGHTS_FILE = None
#WEIGHTS_FILE = os.path.join(DATA_PATH, 'weights1', 'YOLO_small.ckpt')

CLASSES = {'Car': '0', 'Truck' : '1', 'Pedestrian' : '2'}
CLASSES_LIST = ['Car', 'Truck', 'Pedestrian']

FLIPPED = True


#
# model parameter
#
ORI_HEIGHT = 1200

ORI_WIDTH = 1920

IMAGE_SIZE1 = 1024

IMAGE_SIZE2 = 1920

#CELL_SIZE = 7

CELL_SIZE1 = 16

CELL_SIZE2 = 30

BOXES_PER_CELL = 1

ALPHA = 0.1

DISP_CONSOLE = True

OBJECT_SCALE = 5.0
NOOBJECT_SCALE = 0.5
CLASS_SCALE = 2.0
COORD_SCALE = 5.0


#
# solver parameter
#

GPU = ''

LEARNING_RATE = 0.0001

DECAY_STEPS = 30000

DECAY_RATE = 0.1

STAIRCASE = True

BATCH_SIZE = 10

MAX_ITER = 140

SUMMARY_ITER = 1

SAVE_ITER = 1


#
# test parameter
#

THRESHOLD = 0.0

IOU_THRESHOLD = 0.0