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

crop_wとcrop_hが違う値の時にTTAをやると出力がおかしくなるバグを修正 #43

Browse Source
This commit is contained in:
lltcggie 2016-09-11 14:06:32 +09:00
parent bb13740260
commit c7b4a6d4be
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

@ -1050,7 +1050,10 @@ Waifu2x::eWaifu2xError Waifu2x::ReconstructByNet(std::shared_ptr<cNet> net, cons
if (i >= 4) if (i >= 4)
cv::flip(in, in, 1); // <20>‚’¼Ž²”½“] cv::flip(in, in, 1); // <20>‚’¼Ž²”½“]
ret = ProcessNet(net, crop_w, crop_h, use_tta, batch_size, in); const int cw = (rotateNum % 2 == 0) ? crop_w : crop_h;
const int ch = (rotateNum % 2 == 0) ? crop_h : crop_w;
ret = ProcessNet(net, cw, ch, use_tta, batch_size, in);
if (ret != Waifu2x::eWaifu2xError_OK) if (ret != Waifu2x::eWaifu2xError_OK)
return ret; return ret;