diff --git a/waifu2x-caffe-gui/Resource.rc b/waifu2x-caffe-gui/Resource.rc index 04c639a..0b4db75 100644 Binary files a/waifu2x-caffe-gui/Resource.rc and b/waifu2x-caffe-gui/Resource.rc differ diff --git a/waifu2x-caffe-gui/Source.cpp b/waifu2x-caffe-gui/Source.cpp index 5d1d6c0..891383f 100644 --- a/waifu2x-caffe-gui/Source.cpp +++ b/waifu2x-caffe-gui/Source.cpp @@ -110,6 +110,8 @@ private: std::string outputExt; std::string inputFileExt; + bool use_tta; + int crop_size; int batch_size; @@ -249,6 +251,8 @@ private: } } + use_tta = SendMessage(GetDlgItem(dh, IDC_CHECK_TTA), BM_GETCHECK, 0, 0) == BST_CHECKED; + return ret; } @@ -431,7 +435,7 @@ private: Waifu2x::eWaifu2xError ret; Waifu2x w; - ret = w.init(__argc, __argv, mode, noise_level, scale_ratio, model_dir, process, crop_size, batch_size); + ret = w.init(__argc, __argv, mode, noise_level, scale_ratio, model_dir, process, use_tta, crop_size, batch_size); if(ret != Waifu2x::eWaifu2xError_OK) SendMessage(dh, WM_ON_WAIFU2X_ERROR, (WPARAM)&ret, 0); else @@ -547,7 +551,7 @@ private: public: DialogEvent() : dh(nullptr), mode("noise_scale"), noise_level(1), scale_ratio(2.0), model_dir("models/anime_style_art_rgb"), process("gpu"), outputExt("png"), inputFileExt("png:jpg:jpeg:tif:tiff:bmp:tga"), - crop_size(128), batch_size(1), isLastError(false) + use_tta(false), crop_size(128), batch_size(1), isLastError(false) { } diff --git a/waifu2x-caffe-gui/resource.h b/waifu2x-caffe-gui/resource.h index 88b84be..8ee177c 100644 Binary files a/waifu2x-caffe-gui/resource.h and b/waifu2x-caffe-gui/resource.h differ