mirror of
https://github.com/lltcggie/waifu2x-caffe.git
synced 2025-06-26 13:42:48 +00:00
ノイズ除去だけの場合でもscale2.0x_modelを読み込んでいたバグを修正
This commit is contained in:
parent
991974a240
commit
cc711eb870
@ -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";
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user