forked from shinjayne/shinTB
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
37 lines (32 loc) · 1003 Bytes
/
config.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
# Configuration for Model Tuning
config = {
#Dataset
"train_data_xml" : "svt1/train.xml",
"test_data_xml" : "svt1/test.xml" ,
#image size
"image_size" : 300.0,
##################
# In Learning
##################
# Restoring variables from ckpt file
"saved_dir" : "saved",
# Random images Batch size in training
"batch_size" : 1,
# Learning rate until step 4000, 180000, 240000, and after
"learning_rate_list" : [8e-4, 1e-3, 1e-4, 1e-5] ,
# Whether Doing Batch Normalization in training phase
"batch_norm" : True,
# Negative and Positive Ratio in Overlapping
"neg/pos": 3,
########################
# In Default Box Shaping
########################
# Layer boxes
"layer_boxes" : [12, 12, 12, 12, 12, 12] ,
# Default box Ratio at every single Map Point
"box_ratios" : [1.0, 2.0, 3.0, 5.0, 7.0, 10.0],
# Default box Scale minimum
"box_s_min" : 0.1 ,
# Total Box number
"total_boxes" : 23280 ,
}