diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise1_scale2.0x_model.prototxt b/bin/models/upconv_7_anime_style_art_rgb/noise1_scale2.0x_model.prototxt index ac5f684..0373400 100644 --- a/bin/models/upconv_7_anime_style_art_rgb/noise1_scale2.0x_model.prototxt +++ b/bin/models/upconv_7_anime_style_art_rgb/noise1_scale2.0x_model.prototxt @@ -3,7 +3,7 @@ layer { name: "input" type: "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 { name: "conv1_layer" diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise2_scale2.0x_model.prototxt b/bin/models/upconv_7_anime_style_art_rgb/noise2_scale2.0x_model.prototxt index ac5f684..0373400 100644 --- a/bin/models/upconv_7_anime_style_art_rgb/noise2_scale2.0x_model.prototxt +++ b/bin/models/upconv_7_anime_style_art_rgb/noise2_scale2.0x_model.prototxt @@ -3,7 +3,7 @@ layer { name: "input" type: "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 { name: "conv1_layer" diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise3_scale2.0x_model.prototxt b/bin/models/upconv_7_anime_style_art_rgb/noise3_scale2.0x_model.prototxt index ac5f684..0373400 100644 --- a/bin/models/upconv_7_anime_style_art_rgb/noise3_scale2.0x_model.prototxt +++ b/bin/models/upconv_7_anime_style_art_rgb/noise3_scale2.0x_model.prototxt @@ -3,7 +3,7 @@ layer { name: "input" type: "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 { name: "conv1_layer" diff --git a/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.prototxt b/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.prototxt index ac5f684..0373400 100644 --- a/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.prototxt +++ b/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.prototxt @@ -3,7 +3,7 @@ layer { name: "input" type: "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 { name: "conv1_layer" diff --git a/bin/models/upconv_7_photo/noise1_scale2.0x_model.prototxt b/bin/models/upconv_7_photo/noise1_scale2.0x_model.prototxt index ac5f684..0373400 100644 --- a/bin/models/upconv_7_photo/noise1_scale2.0x_model.prototxt +++ b/bin/models/upconv_7_photo/noise1_scale2.0x_model.prototxt @@ -3,7 +3,7 @@ layer { name: "input" type: "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 { name: "conv1_layer" diff --git a/bin/models/upconv_7_photo/noise2_scale2.0x_model.prototxt b/bin/models/upconv_7_photo/noise2_scale2.0x_model.prototxt index ac5f684..0373400 100644 --- a/bin/models/upconv_7_photo/noise2_scale2.0x_model.prototxt +++ b/bin/models/upconv_7_photo/noise2_scale2.0x_model.prototxt @@ -3,7 +3,7 @@ layer { name: "input" type: "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 { name: "conv1_layer" diff --git a/bin/models/upconv_7_photo/noise3_scale2.0x_model.prototxt b/bin/models/upconv_7_photo/noise3_scale2.0x_model.prototxt index ac5f684..0373400 100644 --- a/bin/models/upconv_7_photo/noise3_scale2.0x_model.prototxt +++ b/bin/models/upconv_7_photo/noise3_scale2.0x_model.prototxt @@ -3,7 +3,7 @@ layer { name: "input" type: "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 { name: "conv1_layer" diff --git a/bin/models/upconv_7_photo/scale2.0x_model.prototxt b/bin/models/upconv_7_photo/scale2.0x_model.prototxt index ac5f684..0373400 100644 --- a/bin/models/upconv_7_photo/scale2.0x_model.prototxt +++ b/bin/models/upconv_7_photo/scale2.0x_model.prototxt @@ -3,7 +3,7 @@ layer { name: "input" type: "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 { name: "conv1_layer" diff --git a/common/cNet.cpp b/common/cNet.cpp index 8819c53..080f581 100644 --- a/common/cNet.cpp +++ b/common/cNet.cpp @@ -349,9 +349,6 @@ Waifu2x::eWaifu2xError cNet::SetParameter(caffe::NetParameter ¶m, const std: auto mid = input_layer->mutable_input_param()->mutable_shape(); if (mid->size() != 1 || mid->Mutable(0)->dim_size() != 4) 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++)