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

ノイズ除去と拡大でUpCUnetではなくCUnetが使われていたのを修正 resolved #141

Browse Source
This commit is contained in:
lltcggie 2019-01-16 02:44:33 +09:00
parent cc711eb870
commit 3f26c040f2
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
common/waifu2x.cpp
View File

@ -655,9 +655,12 @@ Waifu2x::eWaifu2xError Waifu2x::Init(const eWaifu2xModelType mode, const int noi
if (ret != Waifu2x::eWaifu2xError_OK)
return ret;
mHasNoiseScaleOnly = info.has_noise_scale && !info.has_noise_only;
mHasNoiseScaleOnly = info.has_noise_scale;
mInputPlane = info.channels;
if (mode == eWaifu2xModelTypeNoise && info.has_noise_only) // ノイズ除去だけかつノイズ除去モデルが存在するのであればノイズ除去スケールモデルは使わないようにする
mHasNoiseScaleOnly = false;
if (mode == eWaifu2xModelTypeNoise || mode == eWaifu2xModelTypeNoiseScale || mode == eWaifu2xModelTypeAutoScale)
{
std::string base_name;