diff --git a/.gitignore b/.gitignore index 3a2238d..902f4de 100644 --- a/.gitignore +++ b/.gitignore @@ -242,4 +242,6 @@ ModelManifest.xml .paket/paket.exe # FAKE - F# Make -.fake/ \ No newline at end of file +.fake/ +*.caffemodel +*.protobin diff --git a/bin/models/anime_style_art/noise1_model.json.caffemodel b/bin/models/anime_style_art/noise1_model.json.caffemodel deleted file mode 100644 index 8cd0c5f..0000000 Binary files a/bin/models/anime_style_art/noise1_model.json.caffemodel and /dev/null differ diff --git a/bin/models/anime_style_art/noise1_model.prototxt b/bin/models/anime_style_art/noise1_model.prototxt new file mode 100644 index 0000000..06adcec --- /dev/null +++ b/bin/models/anime_style_art/noise1_model.prototxt @@ -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 } +} diff --git a/bin/models/anime_style_art/noise2_model.json.caffemodel b/bin/models/anime_style_art/noise2_model.json.caffemodel deleted file mode 100644 index 6121f84..0000000 Binary files a/bin/models/anime_style_art/noise2_model.json.caffemodel and /dev/null differ diff --git a/bin/models/anime_style_art/noise2_model.prototxt b/bin/models/anime_style_art/noise2_model.prototxt new file mode 100644 index 0000000..06adcec --- /dev/null +++ b/bin/models/anime_style_art/noise2_model.prototxt @@ -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 } +} diff --git a/bin/models/anime_style_art/noise3_model.json.caffemodel b/bin/models/anime_style_art/noise3_model.json.caffemodel deleted file mode 100644 index 23fa720..0000000 Binary files a/bin/models/anime_style_art/noise3_model.json.caffemodel and /dev/null differ diff --git a/bin/models/anime_style_art/noise3_model.prototxt b/bin/models/anime_style_art/noise3_model.prototxt new file mode 100644 index 0000000..06adcec --- /dev/null +++ b/bin/models/anime_style_art/noise3_model.prototxt @@ -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 } +} diff --git a/bin/models/anime_style_art/scale2.0x_model.json.caffemodel b/bin/models/anime_style_art/scale2.0x_model.json.caffemodel deleted file mode 100644 index 4eba188..0000000 Binary files a/bin/models/anime_style_art/scale2.0x_model.json.caffemodel and /dev/null differ diff --git a/bin/models/anime_style_art_rgb/noise1_model.json.caffemodel b/bin/models/anime_style_art_rgb/noise1_model.json.caffemodel deleted file mode 100644 index 3585aba..0000000 Binary files a/bin/models/anime_style_art_rgb/noise1_model.json.caffemodel and /dev/null differ diff --git a/bin/models/photo/srcnn.prototxt b/bin/models/anime_style_art_rgb/noise1_model.prototxt similarity index 100% rename from bin/models/photo/srcnn.prototxt rename to bin/models/anime_style_art_rgb/noise1_model.prototxt diff --git a/bin/models/anime_style_art_rgb/noise2_model.json.caffemodel b/bin/models/anime_style_art_rgb/noise2_model.json.caffemodel deleted file mode 100644 index 2fc81c9..0000000 Binary files a/bin/models/anime_style_art_rgb/noise2_model.json.caffemodel and /dev/null differ diff --git a/bin/models/ukbench/srcnn.prototxt b/bin/models/anime_style_art_rgb/noise2_model.prototxt similarity index 100% rename from bin/models/ukbench/srcnn.prototxt rename to bin/models/anime_style_art_rgb/noise2_model.prototxt diff --git a/bin/models/anime_style_art_rgb/noise3_model.json.caffemodel b/bin/models/anime_style_art_rgb/noise3_model.json.caffemodel deleted file mode 100644 index 49eb245..0000000 Binary files a/bin/models/anime_style_art_rgb/noise3_model.json.caffemodel and /dev/null differ diff --git a/bin/models/anime_style_art_rgb/noise3_model.prototxt b/bin/models/anime_style_art_rgb/noise3_model.prototxt new file mode 100644 index 0000000..430ce6e --- /dev/null +++ b/bin/models/anime_style_art_rgb/noise3_model.prototxt @@ -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 } +} diff --git a/bin/models/anime_style_art_rgb/scale2.0x_model.json.caffemodel b/bin/models/anime_style_art_rgb/scale2.0x_model.json.caffemodel deleted file mode 100644 index 8e01725..0000000 Binary files a/bin/models/anime_style_art_rgb/scale2.0x_model.json.caffemodel and /dev/null differ diff --git a/bin/models/photo/noise1_model.json.caffemodel b/bin/models/photo/noise1_model.json.caffemodel deleted file mode 100644 index dd77fd7..0000000 Binary files a/bin/models/photo/noise1_model.json.caffemodel and /dev/null differ diff --git a/bin/models/photo/noise1_model.prototxt b/bin/models/photo/noise1_model.prototxt new file mode 100644 index 0000000..430ce6e --- /dev/null +++ b/bin/models/photo/noise1_model.prototxt @@ -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 } +} diff --git a/bin/models/photo/noise2_model.json.caffemodel b/bin/models/photo/noise2_model.json.caffemodel deleted file mode 100644 index 1e0edcf..0000000 Binary files a/bin/models/photo/noise2_model.json.caffemodel and /dev/null differ diff --git a/bin/models/photo/noise2_model.prototxt b/bin/models/photo/noise2_model.prototxt new file mode 100644 index 0000000..430ce6e --- /dev/null +++ b/bin/models/photo/noise2_model.prototxt @@ -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 } +} diff --git a/bin/models/photo/noise3_model.json.caffemodel b/bin/models/photo/noise3_model.json.caffemodel deleted file mode 100644 index 3b3acb7..0000000 Binary files a/bin/models/photo/noise3_model.json.caffemodel and /dev/null differ diff --git a/bin/models/photo/noise3_model.prototxt b/bin/models/photo/noise3_model.prototxt new file mode 100644 index 0000000..430ce6e --- /dev/null +++ b/bin/models/photo/noise3_model.prototxt @@ -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 } +} diff --git a/bin/models/photo/scale2.0x_model.json.caffemodel b/bin/models/photo/scale2.0x_model.json.caffemodel deleted file mode 100644 index d1b0712..0000000 Binary files a/bin/models/photo/scale2.0x_model.json.caffemodel and /dev/null differ diff --git a/bin/models/photo/scale2.0x_model.prototxt b/bin/models/photo/scale2.0x_model.prototxt new file mode 100644 index 0000000..430ce6e --- /dev/null +++ b/bin/models/photo/scale2.0x_model.prototxt @@ -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 } +} diff --git a/bin/models/ukbench/scale2.0x_model.json.caffemodel b/bin/models/ukbench/scale2.0x_model.json.caffemodel deleted file mode 100644 index d1ceff0..0000000 Binary files a/bin/models/ukbench/scale2.0x_model.json.caffemodel and /dev/null differ diff --git a/bin/models/ukbench/scale2.0x_model.prototxt b/bin/models/ukbench/scale2.0x_model.prototxt new file mode 100644 index 0000000..430ce6e --- /dev/null +++ b/bin/models/ukbench/scale2.0x_model.prototxt @@ -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 } +} diff --git a/bin/models/upconv_7_anime_style_art_rgb/info.json b/bin/models/upconv_7_anime_style_art_rgb/info.json index e336254..a90cca2 100644 --- a/bin/models/upconv_7_anime_style_art_rgb/info.json +++ b/bin/models/upconv_7_anime_style_art_rgb/info.json @@ -1 +1 @@ -{"name":"UP","arch_name":"upconv_7","channels":3,"resize":true,"scale_factor":2,"offset":12} \ No newline at end of file +{"name":"UpRGB","arch_name":"upconv_7","channels":3,"resize":true,"scale_factor":2,"offset":12} \ No newline at end of file diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise1_model.json.caffemodel b/bin/models/upconv_7_anime_style_art_rgb/noise1_model.json.caffemodel deleted file mode 100644 index 8dfaf9e..0000000 Binary files a/bin/models/upconv_7_anime_style_art_rgb/noise1_model.json.caffemodel and /dev/null differ diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise1_model.prototxt b/bin/models/upconv_7_anime_style_art_rgb/noise1_model.prototxt new file mode 100644 index 0000000..430ce6e --- /dev/null +++ b/bin/models/upconv_7_anime_style_art_rgb/noise1_model.prototxt @@ -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 } +} diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise2_model.prototxt b/bin/models/upconv_7_anime_style_art_rgb/noise2_model.prototxt new file mode 100644 index 0000000..430ce6e --- /dev/null +++ b/bin/models/upconv_7_anime_style_art_rgb/noise2_model.prototxt @@ -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 } +} diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise3_model.prototxt b/bin/models/upconv_7_anime_style_art_rgb/noise3_model.prototxt new file mode 100644 index 0000000..430ce6e --- /dev/null +++ b/bin/models/upconv_7_anime_style_art_rgb/noise3_model.prototxt @@ -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 } +} diff --git a/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.json.caffemodel b/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.json.caffemodel deleted file mode 100644 index 1acb14b..0000000 Binary files a/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.json.caffemodel and /dev/null differ