mirror of
https://github.com/lltcggie/waifu2x-caffe.git
synced 2025-06-26 13:42:48 +00:00
CUIでもエラーログの設定を行うようにした
This commit is contained in:
parent
30b8c5fd2e
commit
361c14ccba
@ -5,6 +5,8 @@
|
||||
#include <boost/foreach.hpp>
|
||||
#include <functional>
|
||||
#include <boost/tokenizer.hpp>
|
||||
#include <boost/tokenizer.hpp>
|
||||
#include <glog/logging.h>
|
||||
#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");
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user