From 13870d225b83d785fd6117afcf1479c2b266d5a3 Mon Sep 17 00:00:00 2001 From: nagadomi Date: Mon, 21 Mar 2016 18:27:40 +0900 Subject: [PATCH] Fix condition of eWaifu2xError_FailedParseModelFile --- common/waifu2x.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/common/waifu2x.cpp b/common/waifu2x.cpp index e814b76..0cb9934 100644 --- a/common/waifu2x.cpp +++ b/common/waifu2x.cpp @@ -734,7 +734,7 @@ Waifu2x::eWaifu2xError Waifu2x::SetParameter(caffe::NetParameter ¶m, 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);