-
Notifications
You must be signed in to change notification settings - Fork 0
/
sizeTest_flownet_s.m
125 lines (86 loc) · 2.07 KB
/
sizeTest_flownet_s.m
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
93
94
95
96
97
98
clc, clear, close all
nH = 512;
nV = 384;
%% flownet_S
% conv1
nH1_in = nH;
nV1_in = nV;
p = 3;
s = 2;
f = 7;
c2 = 64;
nV1_out = getOutputSize(nV1_in, p, s, f);
nH2_out = getOutputSize(nH1_in, p, s, f);
fprintf('conv1 , dim: (%d, %d, %d)\n', nV1_out, nH2_out, c2)
% conv2
p = 3;
s = 2;
f = 7;
c3 = 128;
nV2_out = getOutputSize(nV1_out, p, s, f);
nH2_out = getOutputSize(nH2_out, p, s, f);
fprintf('conv2 , dim: (%d, %d, %d)\n', nV2_out, nH2_out, c3)
% conv3
p = 2;
s = 2;
f = 5;
c3_out = 256;
nV3_out = getOutputSize(nV2_out, p, s, f);
nH3_out = getOutputSize(nH2_out, p, s, f);
fprintf('conv3 , dim: (%d, %d, %d)\n', nV3_out, nH3_out, c3_out)
% conv3_1
p = 1;
s = 1;
f = 3;
c3_1_out = 256;
nV3_1_out = getOutputSize(nV3_out, p, s, f);
nH3_1_out = getOutputSize(nH3_out, p, s, f);
fprintf('conv3_1, dim: (%d, %d, %d)\n', nV3_1_out, nH3_1_out, c3_1_out)
% conv4
p = 1;
s = 2;
f = 3;
c4_out = 512;
nV4_out = getOutputSize(nV3_1_out, p, s, f);
nH4_out = getOutputSize(nH3_1_out, p, s, f);
fprintf('conv4 , dim: (%d, %d, %d)\n', nV4_out, nH4_out, c4_out)
% conv4_1
p = 1;
s = 1;
f = 3;
c4_1_out = 512;
nV4_1_out = getOutputSize(nV4_out, p, s, f);
nH4_1_out = getOutputSize(nH4_out, p, s, f);
fprintf('conv4_1, dim: (%d, %d, %d)\n', nV4_1_out, nH4_1_out, c4_1_out)
% conv5
p = 1;
s = 2;
f = 3;
c5_out = 512;
nV5_out = getOutputSize(nV4_1_out, p, s, f);
nH5_out = getOutputSize(nH4_1_out, p, s, f);
fprintf('conv5 , dim: (%d, %d, %d)\n', nV5_out, nH5_out, c5_out)
% conv5_1
p = 1;
s = 1;
f = 3;
c5_1_out = 512;
nV5_1_out = getOutputSize(nV5_out, p, s, f);
nH5_1_out = getOutputSize(nH5_out, p, s, f);
fprintf('conv5_1, dim: (%d, %d, %d)\n', nV5_1_out, nH5_1_out, c5_1_out)
% conv6
p = 1;
s = 2;
f = 3;
c6_out = 1024;
nV6_out = getOutputSize(nV5_1_out, p, s, f);
nH6_out = getOutputSize(nH5_1_out, p, s, f);
fprintf('conv6 , dim: (%d, %d, %d)\n', nV6_out, nH6_out, c6_out)
% conv6_1
p = 1;
s = 1;
f = 3;
c6_1_out = 1024;
nV6_out = getOutputSize(nV5_1_out, p, s, f);
nH6_out = getOutputSize(nH5_1_out, p, s, f);
fprintf('conv6_1, dim: (%d, %d, %d)\n', nV6_out, nH6_out, c6_out)