Skip to content

Commit 00eff75

Browse files
dmitrygolovkindmitrygolovkin
committed
Added 2 new graph models: SRCNN and VDSR
1 parent e2718c5 commit 00eff75

12 files changed

+2221
-0
lines changed

‎caffe_models/srcnn/caffe_model/SRCNN_768x768.prototxt‎

Lines changed: 102 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,102 @@
1+
name: "SRCNN"
2+
input: "data"
3+
input_dim: 1
4+
input_dim: 1
5+
input_dim: 768
6+
input_dim: 768
7+
8+
layer {
9+
name: "conv1"
10+
type: "Convolution"
11+
bottom: "data"
12+
top: "conv1"
13+
param {
14+
lr_mult: 1
15+
}
16+
param {
17+
lr_mult: 0.1
18+
}
19+
convolution_param {
20+
num_output: 64
21+
kernel_size: 9
22+
stride: 1
23+
pad: 0
24+
weight_filler {
25+
type: "gaussian"
26+
std: 0.001
27+
}
28+
bias_filler {
29+
type: "constant"
30+
value: 0
31+
}
32+
}
33+
}
34+
35+
layer {
36+
name: "relu1"
37+
type: "ReLU"
38+
bottom: "conv1"
39+
top: "conv1"
40+
}
41+
42+
layer {
43+
name: "conv2"
44+
type: "Convolution"
45+
bottom: "conv1"
46+
top: "conv2"
47+
param {
48+
lr_mult: 1
49+
}
50+
param {
51+
lr_mult: 0.1
52+
}
53+
convolution_param {
54+
num_output: 32
55+
kernel_size: 1
56+
stride: 1
57+
pad: 0
58+
weight_filler {
59+
type: "gaussian"
60+
std: 0.001
61+
}
62+
bias_filler {
63+
type: "constant"
64+
value: 0
65+
}
66+
}
67+
}
68+
69+
layer {
70+
name: "relu2"
71+
type: "ReLU"
72+
bottom: "conv2"
73+
top: "conv2"
74+
}
75+
76+
layer {
77+
name: "conv3"
78+
type: "Convolution"
79+
bottom: "conv2"
80+
top: "conv3"
81+
param {
82+
lr_mult: 0.1
83+
}
84+
param {
85+
lr_mult: 0.1
86+
}
87+
convolution_param {
88+
num_output: 1
89+
kernel_size: 5
90+
stride: 1
91+
pad: 0
92+
weight_filler {
93+
type: "gaussian"
94+
std: 0.001
95+
}
96+
bias_filler {
97+
type: "constant"
98+
value: 0
99+
}
100+
}
101+
}
102+

‎caffe_models/srcnn/caffe_model/SRCNN_deploynet.prototxt‎

Lines changed: 102 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,102 @@
1+
name: "SRCNN"
2+
input: "data"
3+
input_dim: 1
4+
input_dim: 1
5+
input_dim: 256
6+
input_dim: 256
7+
8+
layer {
9+
name: "conv1"
10+
type: "Convolution"
11+
bottom: "data"
12+
top: "conv1"
13+
param {
14+
lr_mult: 1
15+
}
16+
param {
17+
lr_mult: 0.1
18+
}
19+
convolution_param {
20+
num_output: 64
21+
kernel_size: 9
22+
stride: 1
23+
pad: 0
24+
weight_filler {
25+
type: "gaussian"
26+
std: 0.001
27+
}
28+
bias_filler {
29+
type: "constant"
30+
value: 0
31+
}
32+
}
33+
}
34+
35+
layer {
36+
name: "relu1"
37+
type: "ReLU"
38+
bottom: "conv1"
39+
top: "conv1"
40+
}
41+
42+
layer {
43+
name: "conv2"
44+
type: "Convolution"
45+
bottom: "conv1"
46+
top: "conv2"
47+
param {
48+
lr_mult: 1
49+
}
50+
param {
51+
lr_mult: 0.1
52+
}
53+
convolution_param {
54+
num_output: 32
55+
kernel_size: 1
56+
stride: 1
57+
pad: 0
58+
weight_filler {
59+
type: "gaussian"
60+
std: 0.001
61+
}
62+
bias_filler {
63+
type: "constant"
64+
value: 0
65+
}
66+
}
67+
}
68+
69+
layer {
70+
name: "relu2"
71+
type: "ReLU"
72+
bottom: "conv2"
73+
top: "conv2"
74+
}
75+
76+
layer {
77+
name: "conv3"
78+
type: "Convolution"
79+
bottom: "conv2"
80+
top: "conv3"
81+
param {
82+
lr_mult: 0.1
83+
}
84+
param {
85+
lr_mult: 0.1
86+
}
87+
convolution_param {
88+
num_output: 1
89+
kernel_size: 5
90+
stride: 1
91+
pad: 0
92+
weight_filler {
93+
type: "gaussian"
94+
std: 0.001
95+
}
96+
bias_filler {
97+
type: "constant"
98+
value: 0
99+
}
100+
}
101+
}
102+

‎caffe_models/srcnn/caffe_model/SRCNN_iter_15000000.caffemodel‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:57b02c71d12d4e3e203bafbe7db55ccedc27e2591bf9c230a17f7f26fd1af867
3+
size 33080

‎caffe_models/srcnn/caffe_model/SRCNN_iter_15000000_random_pruned.caffemodel‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:0d211b866f54d30f5ce48d23c036cc477409f1b950a90db26c548d59b6e8b690
3+
size 32997

‎caffe_models/srcnn/caffe_model/notes.txt‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
SRCNN model from: https://github.com/Tonyfy/SRCNN
2+
3+
Pruned Graphs
4+
1. SRCNN_deploynet.prototxt / SRCNN_iter_15000000_random_pruned.caffemodel
5+
- random pruned (60% conv, 85% fc)

‎caffe_models/vdsr/caffe_model/VDSR_Adam.caffemodel‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:9199251105e65ccbe3c720192366383e12fca21afaa71a792595b13eb2d8ae9d
3+
size 2666880

‎caffe_models/vdsr/caffe_model/VDSR_Adam_random_pruned.caffemodel‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:a9dd6eae8d0c50a99e01348bb8262b98802e9c3b3562484a0d9379ab4c165c5b
3+
size 2666806

0 commit comments

Comments
 (0)