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

ノイズ除去でアルファチャンネルが消失するバグを修正 #8

Browse Source
This commit is contained in:
lltcggie 2015-12-31 14:43:27 +09:00
parent 7083f0f785
commit a3ecc985a9
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
common/waifu2x.cpp
View File

@ -1613,12 +1613,15 @@ Waifu2x::eWaifu2xError Waifu2x::AfterReconstructFloatMatProcess(const bool isRec
const double shrinkRatio = scale_ratio / std::pow(2.0, (double)scale2); const double shrinkRatio = scale_ratio / std::pow(2.0, (double)scale2);
cv::Mat alpha; cv::Mat alpha;
if (floatim.channels() == 4 && isReconstructScale) if (floatim.channels() == 4)
{ {
std::vector<cv::Mat> planes; std::vector<cv::Mat> planes;
cv::split(floatim, planes); cv::split(floatim, planes);
if (isReconstructScale)
Reconstruct(false, true, cancel_func, planes[3], alpha); Reconstruct(false, true, cancel_func, planes[3], alpha);
else
alpha = planes[3];
} }
// アルファチャンネルがあったらアルファを付加する // アルファチャンネルがあったらアルファを付加する