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

CUI版でcrop_w,crop_hオプション追加

Browse Source
This commit is contained in:
lltcggie 2016-07-04 21:54:52 +09:00
parent 943b7de3ed
commit 816aa2a758
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
waifu2x-caffe/Source.cpp
View File

@ -131,6 +131,14 @@ int main(int argc, char** argv)
"input image split size", false, "input image split size", false,
128, "int", cmd); 128, "int", cmd);
TCLAP::ValueArg<int> cmdCropWidth("", "crop_w",
"input image split size(width)", false,
128, "int", cmd);
TCLAP::ValueArg<int> cmdCropHeight("", "crop_h",
"input image split size(height)", false,
128, "int", cmd);
TCLAP::ValueArg<int> cmdBatchSizeFile("b", "batch_size", TCLAP::ValueArg<int> cmdBatchSizeFile("b", "batch_size",
"input batch size", false, "input batch size", false,
1, "int", cmd); 1, "int", cmd);
@ -177,6 +185,15 @@ int main(int argc, char** argv)
return 1; return 1;
} }
int crop_w = cmdCropSizeFile.getValue();
int crop_h = cmdCropSizeFile.getValue();
if (cmdCropWidth.isSet())
crop_w = cmdCropWidth.getValue();
if (cmdCropHeight.isSet())
crop_h = cmdCropHeight.getValue();
if (cmdScaleWidth.getValue() > 0) if (cmdScaleWidth.getValue() > 0)
ScaleWidth = cmdScaleWidth.getValue(); ScaleWidth = cmdScaleWidth.getValue();
else if (cmdScaleHeight.getValue() > 0) else if (cmdScaleHeight.getValue() > 0)
@ -382,7 +399,7 @@ int main(int argc, char** argv)
for (const auto &p : file_paths) for (const auto &p : file_paths)
{ {
const Waifu2x::eWaifu2xError ret = w.waifu2x(p.first, p.second, ScaleRatio, ScaleWidth, ScaleHeight, nullptr, const Waifu2x::eWaifu2xError ret = w.waifu2x(p.first, p.second, ScaleRatio, ScaleWidth, ScaleHeight, nullptr,
cmdCropSizeFile.getValue(), cmdCropSizeFile.getValue(), crop_w, crop_h,
cmdOutputQuality.getValue() == -1 ? boost::optional<int>() : cmdOutputQuality.getValue(), cmdOutputDepth.getValue(), use_tta, cmdBatchSizeFile.getValue()); cmdOutputQuality.getValue() == -1 ? boost::optional<int>() : cmdOutputQuality.getValue(), cmdOutputDepth.getValue(), use_tta, cmdBatchSizeFile.getValue());
if (ret != Waifu2x::eWaifu2xError_OK) if (ret != Waifu2x::eWaifu2xError_OK)
{ {