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

TTAモードが正常に動作しなかったバグを修正

Browse Source
This commit is contained in:
lltcggie 2016-07-03 21:40:20 +09:00
parent 6ea9a33fa5
commit 41fdf561d8
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
common/waifu2x.cpp
View File

@ -636,7 +636,7 @@ Waifu2x::eWaifu2xError Waifu2x::ReconstructByNet(std::shared_ptr<cNet> net, cons
cv::Mat reconstruct_image; cv::Mat reconstruct_image;
for (int i = 0; i < 8; i++) for (int i = 0; i < 8; i++)
{ {
cv::Mat in(im); cv::Mat in(im.clone());
const int rotateNum = i % 4; const int rotateNum = i % 4;
RotateClockwise90N(in, rotateNum); RotateClockwise90N(in, rotateNum);
@ -644,7 +644,7 @@ 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, im); ret = ProcessNet(net, crop_w, crop_h, use_tta, batch_size, in);
if (ret != Waifu2x::eWaifu2xError_OK) if (ret != Waifu2x::eWaifu2xError_OK)
return ret; return ret;
@ -660,6 +660,8 @@ Waifu2x::eWaifu2xError Waifu2x::ReconstructByNet(std::shared_ptr<cNet> net, cons
} }
reconstruct_image /= 8.0; reconstruct_image /= 8.0;
im = reconstruct_image;
} }
return Waifu2x::eWaifu2xError_OK; return Waifu2x::eWaifu2xError_OK;