Mirror/waifu2x-caffe
1
0
Fork 0
mirror of https://github.com/lltcggie/waifu2x-caffe.git synced 2025-06-26 05:32:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

各ネットワーク入力の初期サイズをcrop_size 128に適合するようにした

Browse Source
This commit is contained in:
lltcggie 2016-07-05 01:48:53 +09:00
parent da773206bb
commit f43c057150
9 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bin/models/upconv_7_anime_style_art_rgb/noise1_scale2.0x_model.prototxt
View File

@ -3,7 +3,7 @@ layer {
name: "input" name: "input"
type: "Input" type: "Input"
top: "input" top: "input"
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } } input_param { shape: { dim: 1 dim: 3 dim: 156 dim: 156 } }
} }
layer { layer {
name: "conv1_layer" name: "conv1_layer"

2
bin/models/upconv_7_anime_style_art_rgb/noise2_scale2.0x_model.prototxt
View File

@ -3,7 +3,7 @@ layer {
name: "input" name: "input"
type: "Input" type: "Input"
top: "input" top: "input"
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } } input_param { shape: { dim: 1 dim: 3 dim: 156 dim: 156 } }
} }
layer { layer {
name: "conv1_layer" name: "conv1_layer"

2
bin/models/upconv_7_anime_style_art_rgb/noise3_scale2.0x_model.prototxt
View File

@ -3,7 +3,7 @@ layer {
name: "input" name: "input"
type: "Input" type: "Input"
top: "input" top: "input"
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } } input_param { shape: { dim: 1 dim: 3 dim: 156 dim: 156 } }
} }
layer { layer {
name: "conv1_layer" name: "conv1_layer"

2
bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.prototxt
View File

@ -3,7 +3,7 @@ layer {
name: "input" name: "input"
type: "Input" type: "Input"
top: "input" top: "input"
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } } input_param { shape: { dim: 1 dim: 3 dim: 156 dim: 156 } }
} }
layer { layer {
name: "conv1_layer" name: "conv1_layer"

2
bin/models/upconv_7_photo/noise1_scale2.0x_model.prototxt
View File

@ -3,7 +3,7 @@ layer {
name: "input" name: "input"
type: "Input" type: "Input"
top: "input" top: "input"
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } } input_param { shape: { dim: 1 dim: 3 dim: 156 dim: 156 } }
} }
layer { layer {
name: "conv1_layer" name: "conv1_layer"

2
bin/models/upconv_7_photo/noise2_scale2.0x_model.prototxt
View File

@ -3,7 +3,7 @@ layer {
name: "input" name: "input"
type: "Input" type: "Input"
top: "input" top: "input"
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } } input_param { shape: { dim: 1 dim: 3 dim: 156 dim: 156 } }
} }
layer { layer {
name: "conv1_layer" name: "conv1_layer"

2
bin/models/upconv_7_photo/noise3_scale2.0x_model.prototxt
View File

@ -3,7 +3,7 @@ layer {
name: "input" name: "input"
type: "Input" type: "Input"
top: "input" top: "input"
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } } input_param { shape: { dim: 1 dim: 3 dim: 156 dim: 156 } }
} }
layer { layer {
name: "conv1_layer" name: "conv1_layer"

2
bin/models/upconv_7_photo/scale2.0x_model.prototxt
View File

@ -3,7 +3,7 @@ layer {
name: "input" name: "input"
type: "Input" type: "Input"
top: "input" top: "input"
input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } } input_param { shape: { dim: 1 dim: 3 dim: 156 dim: 156 } }
} }
layer { layer {
name: "conv1_layer" name: "conv1_layer"

3
common/cNet.cpp
View File

@ -349,9 +349,6 @@ Waifu2x::eWaifu2xError cNet::SetParameter(caffe::NetParameter &param, const std:
auto mid = input_layer->mutable_input_param()->mutable_shape(); auto mid = input_layer->mutable_input_param()->mutable_shape();
if (mid->size() != 1 || mid->Mutable(0)->dim_size() != 4) if (mid->size() != 1 || mid->Mutable(0)->dim_size() != 4)
return Waifu2x::eWaifu2xError_FailedParseModelFile; return Waifu2x::eWaifu2xError_FailedParseModelFile;
mid->Mutable(0)->set_dim(0, 1);
mid->Mutable(0)->set_dim(2, 142);
mid->Mutable(0)->set_dim(3, 142);
} }
for (int i = 0; i < param.layer_size(); i++) for (int i = 0; i < param.layer_size(); i++)