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

Merge pull request #66 from nagadomi/fix_tta

Browse Source 
ノイズ除去のみのときにTTAが適用できないのを修正
This commit is contained in:
lltcggie 2016-11-20 23:49:31 +09:00 committed by GitHub
commit a9b91ad9f9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/waifu2x.cpp
View File

@ -903,7 +903,7 @@ Waifu2x::eWaifu2xError Waifu2x::ReconstructImage(const double factor, const int
cv::Size_<int> size; cv::Size_<int> size;
image.GetScalePaddingedRGB(im, size, mNoiseNet->GetNetOffset(), OuterPadding, crop_w, crop_h, 1); image.GetScalePaddingedRGB(im, size, mNoiseNet->GetNetOffset(), OuterPadding, crop_w, crop_h, 1);
ret = ProcessNet(mNoiseNet, crop_w, crop_h, use_tta, batch_size, im); ret = ReconstructByNet(mNoiseNet, crop_w, crop_h, use_tta, batch_size, cancel_func, im);
if (ret != Waifu2x::eWaifu2xError_OK) if (ret != Waifu2x::eWaifu2xError_OK)
return ret; return ret;