ノイズ除去だけの場合でもscale2.0x_modelを読み込んでいたバグを修正

This commit is contained in:
lltcggie 2019-01-16 02:32:49 +09:00
parent 991974a240
commit cc711eb870

View File

@ -687,8 +687,8 @@ Waifu2x::eWaifu2xError Waifu2x::Init(const eWaifu2xModelType mode, const int noi
mMaxNetOffset = mNoiseNet->GetNetOffset(); mMaxNetOffset = mNoiseNet->GetNetOffset();
} }
// noise_scaleを持っている場合はαチャンネルの拡大のためにmScaleNetも構築する必要がある // 拡大が必要な場合はαチャンネルの拡大のためにmScaleNetも構築する必要がある
if (info.has_noise_scale || mode == eWaifu2xModelTypeScale || mode == eWaifu2xModelTypeNoiseScale || mode == eWaifu2xModelTypeAutoScale) if (mode == eWaifu2xModelTypeScale || mode == eWaifu2xModelTypeNoiseScale || mode == eWaifu2xModelTypeAutoScale)
{ {
const std::string base_name = "scale2.0x_model"; const std::string base_name = "scale2.0x_model";