diff --git a/waifu2x-caffe/Source.cpp b/waifu2x-caffe/Source.cpp index aa9ee3c..9e0142d 100644 --- a/waifu2x-caffe/Source.cpp +++ b/waifu2x-caffe/Source.cpp @@ -5,6 +5,8 @@ #include #include #include +#include +#include #include "../common/waifu2x.h" @@ -52,6 +54,14 @@ int main(int argc, char** argv) { Waifu2x::init_liblary(); + // Caffeのエラーでないログを保存しないようにする + google::SetLogDestination(google::INFO, ""); + google::SetLogDestination(google::WARNING, ""); + + // Caffeのエラーログを「error_log_〜」に出力 + google::SetLogDestination(google::ERROR, "error_log_"); + google::SetLogDestination(google::FATAL, "error_log_"); + // definition of command line arguments TCLAP::CmdLine cmd("waifu2x reimplementation using Caffe", ' ', "1.0.0");