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

CUIでもTTAを使うときにファイル名が変化するようにした

Browse Source
This commit is contained in:
lltcggie 2015-11-19 02:48:30 +09:00
parent 08555f0864
commit fe09583367
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
waifu2x-caffe/Source.cpp
View File

@ -136,6 +136,8 @@ int main(int argc, char** argv)
if (outputExt.length() > 0 && outputExt[0] != '.') if (outputExt.length() > 0 && outputExt[0] != '.')
outputExt = "." + outputExt; outputExt = "." + outputExt;
const bool use_tta = cmdTTALevel.getValue() == 1;
std::vector<std::pair<std::string, std::string>> file_paths; std::vector<std::pair<std::string, std::string>> file_paths;
if (boost::filesystem::is_directory(input_path)) // input_pathがフォルダならそのディレクトリ以下の画像ファイルを一括変換 if (boost::filesystem::is_directory(input_path)) // input_pathがフォルダならそのディレクトリ以下の画像ファイルを一括変換
{ {
@ -150,6 +152,8 @@ int main(int argc, char** argv)
const std::string &mode = cmdMode.getValue(); const std::string &mode = cmdMode.getValue();
if (mode.find("noise") != mode.npos || mode.find("auto_scale") != mode.npos) if (mode.find("noise") != mode.npos || mode.find("auto_scale") != mode.npos)
addstr += "(Level" + std::to_string(cmdNRLevel.getValue()) + ")"; addstr += "(Level" + std::to_string(cmdNRLevel.getValue()) + ")";
if (use_tta)
addstr += "(tta)";
if (mode.find("scale") != mode.npos) if (mode.find("scale") != mode.npos)
addstr += "(x" + std::to_string(cmdScaleRatio.getValue()) + ")"; addstr += "(x" + std::to_string(cmdScaleRatio.getValue()) + ")";
@ -244,6 +248,8 @@ int main(int argc, char** argv)
std::string &mode = cmdMode.getValue(); std::string &mode = cmdMode.getValue();
if (mode.find("noise") != mode.npos || mode.find("auto_scale") != mode.npos) if (mode.find("noise") != mode.npos || mode.find("auto_scale") != mode.npos)
outputFileName = outputFileName + "(Level" + std::to_string(cmdNRLevel.getValue()) + ")"; outputFileName = outputFileName + "(Level" + std::to_string(cmdNRLevel.getValue()) + ")";
if (use_tta)
outputFileName += "(tta)";
if (mode.find("scale") != mode.npos) if (mode.find("scale") != mode.npos)
outputFileName = outputFileName + "(x" + std::to_string(cmdScaleRatio.getValue()) + ")"; outputFileName = outputFileName + "(x" + std::to_string(cmdScaleRatio.getValue()) + ")";
outputFileName += outputExt; outputFileName += outputExt;
@ -254,7 +260,7 @@ int main(int argc, char** argv)
Waifu2x::eWaifu2xError ret; Waifu2x::eWaifu2xError ret;
Waifu2x w; Waifu2x w;
ret = w.init(argc, argv, cmdMode.getValue(), cmdNRLevel.getValue(), cmdScaleRatio.getValue(), cmdModelPath.getValue(), cmdProcess.getValue(), cmdTTALevel.getValue() == 1, ret = w.init(argc, argv, cmdMode.getValue(), cmdNRLevel.getValue(), cmdScaleRatio.getValue(), cmdModelPath.getValue(), cmdProcess.getValue(), use_tta,
cmdCropSizeFile.getValue(), cmdBatchSizeFile.getValue()); cmdCropSizeFile.getValue(), cmdBatchSizeFile.getValue());
switch (ret) switch (ret)
{ {