-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathconfig.py
More file actions
92 lines (78 loc) · 2.17 KB
/
config.py
File metadata and controls
92 lines (78 loc) · 2.17 KB
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
84
85
86
87
88
89
90
91
92
#########################################################################
# MODEL PARAMETERS #
#########################################################################
# version (0 for VGG and 1 for ResNet)
version = 0
# batch size
video_b_s = 2
image_b_s = 20
# number of frames
num_frames = 5
# origin shape c
normal_shape_c = 640
# origin shape r
normal_shape_r = 360
# number of cols of input images
shape_c = 320
if version == 1:
# number of rows of input images
shape_r = 256
# number of rows of model outputs
shape_r_out = 64
# number of cols of model outputs
shape_c_out = 80
else:
# number of rows of input images
shape_r = 256#240
# number of cols of model outputs
shape_c_out = 160
# number of rows of model outputs
shape_r_out = 128#120
# number of rows of attention
shape_r_attention = 64
# number of cols of attention
shape_c_attention = 80
# number of rows of downsampled maps
shape_r_gt = 32
# number of cols of downsampled maps
shape_c_gt = 40
# final upsampling factor
upsampling_factor = 16
# number of epochs
nb_epoch = 10
# number of timestep
nb_timestep = 4
# number of learned priors
nb_gaussian = 16
# path of continuous saliency map
maps_path = '/maps/'
# path of fixation maps
fixs_path = '/fixation/maps/'
# path of images
frames_path = '/images/'
# number of training videos
nb_train = 100
# number of validation videos
nb_videos_val = 150
# number of iterations for training using pytorch
nb_train_pt = 100000
# number of small test batch while training
nb_small_test_batch = 10
#########################################################################
# TRAINING SETTINGS #
#########################################################################
# path of training videos
videos_train_paths = ['./train/']
# path of validation videos
videos_val_paths = ['./test/']
# path of validation videos
videos_test_path = './test/'
videos_small_test_path = './train/'
output_path = './output/'
# path of training maps
maps_path = '/maps/'
# path of training fixation maps
fixs_path = '/fixation/maps/'
frames_path = '/images/'
# path of training images
imgs_path = 'D:/code/attention/staticimages/training/'