mirror of
https://github.com/lltcggie/waifu2x-caffe.git
synced 2025-06-25 21:22:47 +00:00
モデルを整理
This commit is contained in:
parent
9052df543f
commit
e88c4a4ca9
4
.gitignore
vendored
4
.gitignore
vendored
@ -242,4 +242,6 @@ ModelManifest.xml
|
||||
.paket/paket.exe
|
||||
|
||||
# FAKE - F# Make
|
||||
.fake/
|
||||
.fake/
|
||||
*.caffemodel
|
||||
*.protobin
|
||||
|
Binary file not shown.
194
bin/models/anime_style_art/noise1_model.prototxt
Normal file
194
bin/models/anime_style_art/noise1_model.prototxt
Normal file
@ -0,0 +1,194 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 1 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 1
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
Binary file not shown.
194
bin/models/anime_style_art/noise2_model.prototxt
Normal file
194
bin/models/anime_style_art/noise2_model.prototxt
Normal file
@ -0,0 +1,194 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 1 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 1
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
Binary file not shown.
194
bin/models/anime_style_art/noise3_model.prototxt
Normal file
194
bin/models/anime_style_art/noise3_model.prototxt
Normal file
@ -0,0 +1,194 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 1 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 1
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
group: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
187
bin/models/anime_style_art_rgb/noise3_model.prototxt
Normal file
187
bin/models/anime_style_art_rgb/noise3_model.prototxt
Normal file
@ -0,0 +1,187 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 3
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
Binary file not shown.
Binary file not shown.
187
bin/models/photo/noise1_model.prototxt
Normal file
187
bin/models/photo/noise1_model.prototxt
Normal file
@ -0,0 +1,187 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 3
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
Binary file not shown.
187
bin/models/photo/noise2_model.prototxt
Normal file
187
bin/models/photo/noise2_model.prototxt
Normal file
@ -0,0 +1,187 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 3
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
Binary file not shown.
187
bin/models/photo/noise3_model.prototxt
Normal file
187
bin/models/photo/noise3_model.prototxt
Normal file
@ -0,0 +1,187 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 3
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
Binary file not shown.
187
bin/models/photo/scale2.0x_model.prototxt
Normal file
187
bin/models/photo/scale2.0x_model.prototxt
Normal file
@ -0,0 +1,187 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 3
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
Binary file not shown.
187
bin/models/ukbench/scale2.0x_model.prototxt
Normal file
187
bin/models/ukbench/scale2.0x_model.prototxt
Normal file
@ -0,0 +1,187 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 3
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
@ -1 +1 @@
|
||||
{"name":"UP","arch_name":"upconv_7","channels":3,"resize":true,"scale_factor":2,"offset":12}
|
||||
{"name":"UpRGB","arch_name":"upconv_7","channels":3,"resize":true,"scale_factor":2,"offset":12}
|
Binary file not shown.
187
bin/models/upconv_7_anime_style_art_rgb/noise1_model.prototxt
Normal file
187
bin/models/upconv_7_anime_style_art_rgb/noise1_model.prototxt
Normal file
@ -0,0 +1,187 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 3
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
187
bin/models/upconv_7_anime_style_art_rgb/noise2_model.prototxt
Normal file
187
bin/models/upconv_7_anime_style_art_rgb/noise2_model.prototxt
Normal file
@ -0,0 +1,187 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 3
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
187
bin/models/upconv_7_anime_style_art_rgb/noise3_model.prototxt
Normal file
187
bin/models/upconv_7_anime_style_art_rgb/noise3_model.prototxt
Normal file
@ -0,0 +1,187 @@
|
||||
name: "srcnn"
|
||||
layer {
|
||||
name: "input"
|
||||
type: "Input"
|
||||
top: "input"
|
||||
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
|
||||
}
|
||||
layer {
|
||||
name: "conv1_layer"
|
||||
type: "Convolution"
|
||||
bottom: "input"
|
||||
top: "conv1"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv1_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv1"
|
||||
top: "conv1"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv1"
|
||||
top: "conv2"
|
||||
convolution_param {
|
||||
num_output: 32
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv2_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv2"
|
||||
top: "conv2"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv2"
|
||||
top: "conv3"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv3_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv3"
|
||||
top: "conv3"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv3"
|
||||
top: "conv4"
|
||||
convolution_param {
|
||||
num_output: 64
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv4_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv4"
|
||||
top: "conv4"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv4"
|
||||
top: "conv5"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv5_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv5"
|
||||
top: "conv5"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv5"
|
||||
top: "conv6"
|
||||
convolution_param {
|
||||
num_output: 128
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv6_relu_layer"
|
||||
type: "ReLU"
|
||||
bottom: "conv6"
|
||||
top: "conv6"
|
||||
relu_param {
|
||||
negative_slope: 0.1
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "conv7_layer"
|
||||
type: "Convolution"
|
||||
bottom: "conv6"
|
||||
top: "conv7"
|
||||
convolution_param {
|
||||
num_output: 3
|
||||
kernel_size: 3
|
||||
stride: 1
|
||||
weight_filler {
|
||||
type: "gaussian"
|
||||
std: 0.01
|
||||
}
|
||||
}
|
||||
}
|
||||
layer {
|
||||
name: "target"
|
||||
type: "MemoryData"
|
||||
top: "target"
|
||||
top: "dummy_label2"
|
||||
memory_data_param {
|
||||
batch_size: 1
|
||||
channels: 1
|
||||
height: 142
|
||||
width: 142
|
||||
}
|
||||
include: { phase: TRAIN }
|
||||
}
|
||||
layer {
|
||||
name: "loss"
|
||||
type: "EuclideanLoss"
|
||||
bottom: "conv7"
|
||||
bottom: "target"
|
||||
top: "loss"
|
||||
include: { phase: TRAIN }
|
||||
}
|
Binary file not shown.
Loading…
x
Reference in New Issue
Block a user