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

waifu2x()でコールバック関数を設定しないとエラーになるバグを修正

Browse Source
This commit is contained in:
lltcggie 2015-06-01 01:23:13 +09:00
parent 91c3d4c4b1
commit b930ecfaa2
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
common/waifu2x.cpp
View File

@ -465,9 +465,10 @@ eWaifu2xError waifu2x(int argc, char** argv, const std::vector<InputOutputPathPa
int fileCount = 0;
for (const auto &p : file_paths)
{
progress_func(file_paths.size(), fileCount);
if (progress_func)
progress_func(file_paths.size(), fileCount);
if (cancel_func())
if (cancel_func && cancel_func())
return eWaifu2xError_Cancel;
const auto &input_file = p.first;
@ -509,7 +510,7 @@ eWaifu2xError waifu2x(int argc, char** argv, const std::vector<InputOutputPathPa
im = im(cv::Rect(offset, offset, image_size.width, image_size.height));
}
if (cancel_func())
if (cancel_func && cancel_func())
return eWaifu2xError_Cancel;
const int scale2 = ceil(log2(scale_ratio));
@ -538,7 +539,7 @@ eWaifu2xError waifu2x(int argc, char** argv, const std::vector<InputOutputPathPa
continue;
}
if (cancel_func())
if (cancel_func && cancel_func())
return eWaifu2xError_Cancel;
// 再構築した輝度画像とCreateZoomColorImage()で作成した色情報をマージして通常の画像に変換し、書き込む
@ -577,7 +578,8 @@ eWaifu2xError waifu2x(int argc, char** argv, const std::vector<InputOutputPathPa
fileCount++;
}
progress_func(file_paths.size(), fileCount);
if (progress_func)
progress_func(file_paths.size(), fileCount);
return eWaifu2xError_OK;
}