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

Fix condition of eWaifu2xError_FailedParseModelFile

Browse Source
This commit is contained in:
nagadomi 2016-03-21 18:27:40 +09:00 committed by lltcggie
parent 2dbf72d191
commit 13870d225b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/waifu2x.cpp
View File

@ -734,7 +734,7 @@ Waifu2x::eWaifu2xError Waifu2x::SetParameter(caffe::NetParameter &param, const s
{
auto input_layer = param.mutable_layer(0);
auto mid = input_layer->mutable_input_param()->mutable_shape();
if (mid->size() > 0 && mid->Mutable(0)->dim_size() != 4)
if (mid->size() != 1 || mid->Mutable(0)->dim_size() != 4)
return eWaifu2xError_FailedParseModelFile;
mid->Mutable(0)->set_dim(0, batch_size);
mid->Mutable(0)->set_dim(2, input_block_size);