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

cNet::LoadParameterFromJson()の不要なチェックを外した

Browse Source
This commit is contained in:
lltcggie 2018-11-23 18:28:23 +09:00
parent 7d1973c314
commit a96d22b4f4
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
common/cNet.cpp
View File

@ -456,9 +456,6 @@ Waifu2x::eWaifu2xError cNet::LoadParameterFromJson(const boost::filesystem::path
return Waifu2x::eWaifu2xError_FailedParseModelFile; return Waifu2x::eWaifu2xError_FailedParseModelFile;
} }
if (d.Size() != 7)
return Waifu2x::eWaifu2xError_FailedParseModelFile;
int inputPlane = 0; int inputPlane = 0;
int outputPlane = 0; int outputPlane = 0;
try try
@ -477,9 +474,6 @@ Waifu2x::eWaifu2xError cNet::LoadParameterFromJson(const boost::filesystem::path
if (inputPlane != outputPlane) if (inputPlane != outputPlane)
return Waifu2x::eWaifu2xError_FailedParseModelFile; return Waifu2x::eWaifu2xError_FailedParseModelFile;
//if (param.layer_size() < 17)
// return Waifu2x::eWaifu2xError_FailedParseModelFile;
std::vector<boost::shared_ptr<caffe::Layer<float>>> list; std::vector<boost::shared_ptr<caffe::Layer<float>>> list;
auto &v = mNet->layers(); auto &v = mNet->layers();
for (auto &l : v) for (auto &l : v)