Skip to content

Commit 37d0348

Browse files
kevinlin311twkevinlin311tw
committed
code update
1 parent 434f08b commit 37d0348

31 files changed

Lines changed: 39890 additions & 15 deletions

‎examples/SSDH/CIFAR10-32/KevinNet_CIFAR10_32_deploy.prototxt‎

Lines changed: 212 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,212 @@
1+
name: "KevinNet_CIFAR10"
2+
input: "data"
3+
input_dim: 10
4+
input_dim: 3
5+
input_dim: 227
6+
input_dim: 227
7+
layers {
8+
name: "conv1"
9+
type: CONVOLUTION
10+
bottom: "data"
11+
top: "conv1"
12+
convolution_param {
13+
num_output: 96
14+
kernel_size: 11
15+
stride: 4
16+
}
17+
}
18+
layers {
19+
name: "relu1"
20+
type: RELU
21+
bottom: "conv1"
22+
top: "conv1"
23+
}
24+
layers {
25+
name: "pool1"
26+
type: POOLING
27+
bottom: "conv1"
28+
top: "pool1"
29+
pooling_param {
30+
pool: MAX
31+
kernel_size: 3
32+
stride: 2
33+
}
34+
}
35+
layers {
36+
name: "norm1"
37+
type: LRN
38+
bottom: "pool1"
39+
top: "norm1"
40+
lrn_param {
41+
local_size: 5
42+
alpha: 0.0001
43+
beta: 0.75
44+
}
45+
}
46+
layers {
47+
name: "conv2"
48+
type: CONVOLUTION
49+
bottom: "norm1"
50+
top: "conv2"
51+
convolution_param {
52+
num_output: 256
53+
pad: 2
54+
kernel_size: 5
55+
group: 2
56+
}
57+
}
58+
layers {
59+
name: "relu2"
60+
type: RELU
61+
bottom: "conv2"
62+
top: "conv2"
63+
}
64+
layers {
65+
name: "pool2"
66+
type: POOLING
67+
bottom: "conv2"
68+
top: "pool2"
69+
pooling_param {
70+
pool: MAX
71+
kernel_size: 3
72+
stride: 2
73+
}
74+
}
75+
layers {
76+
name: "norm2"
77+
type: LRN
78+
bottom: "pool2"
79+
top: "norm2"
80+
lrn_param {
81+
local_size: 5
82+
alpha: 0.0001
83+
beta: 0.75
84+
}
85+
}
86+
layers {
87+
name: "conv3"
88+
type: CONVOLUTION
89+
bottom: "norm2"
90+
top: "conv3"
91+
convolution_param {
92+
num_output: 384
93+
pad: 1
94+
kernel_size: 3
95+
}
96+
}
97+
layers {
98+
name: "relu3"
99+
type: RELU
100+
bottom: "conv3"
101+
top: "conv3"
102+
}
103+
layers {
104+
name: "conv4"
105+
type: CONVOLUTION
106+
bottom: "conv3"
107+
top: "conv4"
108+
convolution_param {
109+
num_output: 384
110+
pad: 1
111+
kernel_size: 3
112+
group: 2
113+
}
114+
}
115+
layers {
116+
name: "relu4"
117+
type: RELU
118+
bottom: "conv4"
119+
top: "conv4"
120+
}
121+
layers {
122+
name: "conv5"
123+
type: CONVOLUTION
124+
bottom: "conv4"
125+
top: "conv5"
126+
convolution_param {
127+
num_output: 256
128+
pad: 1
129+
kernel_size: 3
130+
group: 2
131+
}
132+
}
133+
layers {
134+
name: "relu5"
135+
type: RELU
136+
bottom: "conv5"
137+
top: "conv5"
138+
}
139+
layers {
140+
name: "pool5"
141+
type: POOLING
142+
bottom: "conv5"
143+
top: "pool5"
144+
pooling_param {
145+
pool: MAX
146+
kernel_size: 3
147+
stride: 2
148+
}
149+
}
150+
layers {
151+
name: "fc6"
152+
type: INNER_PRODUCT
153+
bottom: "pool5"
154+
top: "fc6"
155+
inner_product_param {
156+
num_output: 4096
157+
}
158+
}
159+
layers {
160+
name: "relu6"
161+
type: RELU
162+
bottom: "fc6"
163+
top: "fc6"
164+
}
165+
layers {
166+
name: "drop6"
167+
type: DROPOUT
168+
bottom: "fc6"
169+
top: "fc6"
170+
dropout_param {
171+
dropout_ratio: 0.5
172+
}
173+
}
174+
layers {
175+
name: "fc7"
176+
type: INNER_PRODUCT
177+
bottom: "fc6"
178+
top: "fc7"
179+
inner_product_param {
180+
num_output: 4096
181+
}
182+
}
183+
layers {
184+
name: "relu7"
185+
type: RELU
186+
bottom: "fc7"
187+
top: "fc7"
188+
}
189+
layers {
190+
name: "drop7"
191+
type: DROPOUT
192+
bottom: "fc7"
193+
top: "fc7"
194+
dropout_param {
195+
dropout_ratio: 0.5
196+
}
197+
}
198+
layers {
199+
name: "fc8_kevin"
200+
type: INNER_PRODUCT
201+
bottom: "fc7"
202+
top: "fc8_kevin"
203+
inner_product_param {
204+
num_output: 32
205+
}
206+
}
207+
layers {
208+
name: "fc8_kevin_encode"
209+
bottom: "fc8_kevin"
210+
top: "fc8_kevin_encode"
211+
type: SIGMOID
212+
}

‎examples/SSDH/CIFAR10-32/KevinNet_CIFAR10_32_solver.prototxt‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
train_net: "KevinNet_CIFAR10_32_train.prototxt"
2+
test_net: "KevinNet_CIFAR10_32_test.prototxt"
3+
test_iter: 100
4+
test_interval: 100
5+
base_lr: 0.001
6+
lr_policy: "step"
7+
gamma: 0.1
8+
stepsize: 25000
9+
display: 20
10+
max_iter: 50000
11+
momentum: 0.9
12+
weight_decay: 0.0005
13+
snapshot: 2000
14+
snapshot_prefix: "KevinNet_CIFAR10_32"

0 commit comments

Comments
 (0)